install_env.sh

#!/bin/bash
# 
# Installer for package
# 
# Run: ./install_env.sh

# Create conda env
source ~/miniconda3/etc/profile.d/conda.sh
conda env create -f environment.yml
conda activate diffefwi
conda env list
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117
pip3 install --use-pep517 -e .
pip3 install packaging
pip3 install flash-attn --no-build-isolation
echo 'Created and activated environment:' $(which python)

# Check cupy works as expected
echo 'Checking torch version and GPU'
conda activate diffefwi
python -c 'import torch; print(torch.__version__);  print(torch.cuda.get_device_name(torch.cuda.current_device())); print(torch.ones(10).to("cuda:0"))'
echo 'Done!'