AllianceCan

Interactive jobs with GPU

salloc --account=def-XXX --time=4:0:0 --gres=gpu:1 --ntasks=1 --mem=80G # 1 CPUs

salloc --account=def-XXX --time=4:0:0 --gres=gpu:1 --nodes=2 --mem=80G # 2 CPUs

Example of creating and setting virtual environment

# On beluga Mar 8, 2023

module load python/3.9

source ~/ptorch-gpu/bin/activate

pip install --no-index torch torchvision torchtext torchaudio

pip install ipykernel ipython 

pip install nvidia-pyindex

pip install nvidia-cuda-nvcc

# distance metrics

pip install --upgrade jinja2

pip install -q kaolin==0.13.0 -f https://nvidia-kaolin.s3.us-east-2.amazonaws.com/torch-1.13.1_cu116.html 

# mesh

pip install volume2mesh trimesh pyvista 

# Kernel updated on May 26, 2023

python -m ipykernel install --user --name ipykernel-py3.10 --display-name "PT-gpu py3.10"

# Graham, May 23, 2023

virtualenv --no-download name_of_venv

source ~/name_of_venv/bin/activate

module load python/3.8.10 ipython-kernel/3.8 gentoo/2020 /StdEnv/2020 cudacore/.11.4.2 cuda cudnn

pip3 install --no-index tensorflow tqdm plyfile SimpleITK==2.2.1 

pip3 install --no-index torch torchvision

python -m ipykernel install --user --name ipykernel-py3.9 --display-name "ptorch-gpu py3.9"