PROJECTNAME=diffusion

rm -rf ~/packages/$PROJECTNAME
mkdir ~/packages/$PROJECTNAME

# pip install --ignore-installed -U git+https://github.com/qubvel/segmentation_models.pytorch

module purge
module load python3-DL/torch/1.12.1-cuda10.2
export PYTHONUSERBASE=~/packages/$PROJECTNAME
PATH=$PATH:~/packages/$PROJECTNAME/bin
export PATH

pip install --upgrade pip

pip install numpy --user --no-cache-dir
pip install wandb==0.13.11 --user --no-cache-dir
pip install pytorch_lightning==2.0.1.post0 --user --no-cache-dir
pip install scikit-image --user --no-cache-dir
pip install jsonargparse --user --no-cache-dir
pip install torchtyping --user --no-cache-dir
pip install pandas --user --no-cache-dir
pip install lmdb --user --no-cache-dir
pip install pyarrow --user --no-cache-dir
pip install imageio --user --no-cache-dir
pip install matplotlib --user --no-cache-dir
pip install torchinfo --user --no-cache-dir
pip install fairscale --user --no-cache-dir
pip install torchmetrics[image] --user --no-cache-dir

pip install hydra-core==1.2.0 --user --no-cache-dir
pip install Pillow --user --no-cache-dir
pip install segmentation-models-pytorch --user --no-cache-dir
pip install scipy~=1.9.1 --user --no-cache-dir
pip install einops~=0.5.0 --user --no-cache-dir

pip install ml-collections --user --no-cache-dir
pip install kornia --user --no-cache-dir

pip install jaxtyping --user --no-cache-dir
pip install nibabel --user --no-cache-dir
pip install beartype --user --no-cache-dir

pip install tfrecord==1.14.1 --user --no-cache-dir

pip install pandas --user --no-cache-dir
pip install imageio --user --no-cache-dir
pip install torchio --user --no-cache-dir
pip install fsspec gcsfs --user --no-cache-dir

pip install fsspec plotly --user --no-cache-dir

pip install pytorch_memlab --user --no-cache-dir
pip install diffusers --user --no-cache-dir
pip install transformers --user --no-cache-dir

pip install lpips --user --no-cache-dir