User:Oleaj@uab.edu

Notes To Self:

Use module load cuda10.1/toolkit otherwise you don't get the nvidia drivers. When you install pytorch make sure it is for the correct cuda version.

python import torch print('cuda.version: ', torch.cuda.version)

The above will show the version of cuda pytorch expects.

Use conda environments rather than venv. conda environments can be added to juypter notebook and you have control over the python version.