Skip to content

Install directions for PyTorch assume you have a working Conda install.

Load CUDA

module load gcc cuda

Install

conda install pytorch::pytorch-cuda

Confirm CUDA is enabled

Confirm your pytorch install is working and CUDA is available. Performance is terrible if CUDA is not enabled.

python

Then enter the following code

import torch
torch.cuda.is_available()