运行和验证

操作步骤

1. 使用PuTTY工具，以root用户登录服务器。
2. 执行以下命令启动Docker。

   docker import pytorch_cuda.tar pytorch/cuda:v3
   docker run --gpus all  -it pytorch/cuda:v3 bash

3. 执行以下命令配置环境变量。

   export PATH=/root/anaconda3/bin:$PATH
   export LD_LIBRARY_PATH=/usr/local/cuda/lib64

4. 执行以下命令启动Python3。

   python3
   import torch
   torch.cuda.is_available()