...
See the README files located in /usr/local/pytorch/ (and subdirectories for different cuda versions) for more version options available to install. For a specific combination, you can install one of the artemis-built wheels in a virtual environment. For example:
| Code Block | ||
|---|---|---|
| ||
module load python/3.9.15 magma/2.5.3 openmpi-gcc/3.1.5 cuda/10.2.89
virtualenv --system-site-packages /project/<YOURPROJECT>/torchenv
source /project/<YOURPROJECT>/torchenv/bin/activate
pip install /usr/local/pytorch/cuda10.2/torch-1.12.0a0+git664058f-cp39-cp39-linux_x86_64.whl
python testGPU.py |
*NOTE* You will have to run the install in an interactive job so the installation has enough cpu/ram to complete. (See TensorFlow documentation for similar configuration).
...