Setup Conda Environment

How To Setup a PyTorch Conda Environment on Plano Slurm Cluster

Allocate an Ubuntu 8GPU MI210 job from the partition 1CN128C8G2H_2IB_MI210_Ubuntu22

salloc --exclusive --mem=0 --gres=gpu:8 -p 1CN128C8G2H_2IB_MI210_Ubuntu22

In the allocated session, load ROCm 6.1.2 Environment

module load rocm-6.1.2

Load anaconda3 modulefile 4.12.0

module load anaconda3/4.12.0

Source the conda.sh file

. $CONDA_ROOT/etc/profile.d/conda.sh

Create conda environment named pt-stable with python 3.8

conda create -n pt-stable python=3.8 -y 

Activate the conda environment that was just created

conda activate pt-stable

Install Stable Release of Pytorch https://pytorch.org/get-started/locally/

pip3 --no-cache-dir install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm6.1.2