Running and Verification
Procedure
- Use PuTTY to log in to the server as the root user.
- Start Docker.
docker import pytorch_cuda.tar pytorch/cuda:v3 docker run --gpus all -it pytorch/cuda:v3 bash
- Set the environment variables.
export PATH=/root/anaconda3/bin:$PATH export LD_LIBRARY_PATH=/usr/local/cuda/lib64
- Start Python 3.
python3 import torch torch.cuda.is_available()

Parent topic: PyTorch 1.9 Porting Guide (CentOS 8.2)