一、安装CUDA Toolkit
查看电脑cuda版本:
方法一:NVIDIA Control Panel->左下角系统信息->组件->NVCUDA的版本。
方法二:cmd中输入nvidia-smi->查看CUDA Version。
下载CUDA toolkit。
官方链接:NVIDIA Developer。
下载比CUDA版本相同或更低的即可。
安装的时候可以修改路径,选择精简安装。
输入 nvcc -V 检查是否安装成功。
二、安装cuDNN(可选)
这是一个加速器,可以不装。
与CUDA toolkit相对应的版本即可。
官方链接:https://developer.nvidia.com/rdp/cudnn-archive。
下载.zip文件,解压后分别放到CUDA的安装目录下即可。(CUDA一般在C://Program Files/NVDIA GPU Computing Toolkit/CUDA/中)
三、安装torch-gpu
官网:Previous PyTorch Versions | PyTorch
1. 设置镜像源:
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
2. 选择与CUDA toolkit相对应的版本进行下载。