CUDA Samples:
# REMINDER: do not do this in your home directory - thanks! test -d /srv/public/$USER || mkdir /srv/public/$USER cd /srv/public/$USER git clone https://github.com/NVIDIA/cuda-samples.git cd cuda-samples/Samples/1_Utilities/deviceQuery make clean make deviceQuery ./deviceQuery
PyTorch:
# REMINDER: do not do this in your home directory - thanks! test -d /srv/public/$USER || mkdir /srv/public/$USER cd /srv/public/$USER # create python3 virtual environment python3 -m venv test-env # enter it source test-env/bin/activate # now install pytorch pip3 install torch # latest and preview: https://pytorch.org/get-started/locally/ # previous: https://pytorch.org/get-started/previous-versions/ python3 import torch # check if NVIDIA driver and GPU are supported torch.cuda.is_available() # should return True torch.zeros(1).cuda() # should return tensor([0.], device='cuda:0')