Environments

Pytorch Geometric

List of commands to install pytorch geometric on our workstation. On O2 run the same commands but module load gcc/9.2.0 cuda/11.7 first.

conda create -n torch_env python=3.8

conda activate torch_env

conda install pytorch=1.12  pytorch-cuda=11.7 -c pytorch -c nvidia

pip install torch-scatter -f https://data.pyg.org/whl/torch-1.12.0+cu116.html

pip install torch-sparse -f https://data.pyg.org/whl/torch-1.12.0+cu116.html

pip install torch-geometric

pip install torch-cluster -f https://data.pyg.org/whl/torch-1.12.0+cu116.html

pip install torch-spline-conv -f https://data.pyg.org/whl/torch-1.12.0+cu116.html