检查服务器CUDA状态

检查CUDA相关配置

1. 检查GPU驱动

nvidia-smi

2. 检查CUDA版本

nvcc -V

若未安装CUDA会报错,若已安装会显示CUDA版本。

3. 检查Pytorch是否可用

import torch
torch.cuda.is_available()

若运行失败则torch未安装，若显示True则Pytorch可以用GPU

配置CUDA

1. 下载CUDA
   下载需要的CUDA https://developer.nvidia.com/cuda-toolkit-archive

2. 安装CUDA
   双击exe安装无脑下一步即可，若已经有驱动可以不勾选驱动。

安装Pytorch

1. 下载与python版本、CUDA版本、系统类型匹配的Pytorch轮子包 https://download.pytorch.org/whl/torch/

2. 下载到本地以后使用pip安装即可

pip install 本地路径.whl