1.安装NVDIA Display Driver
无需安装多余组件,自定义安装目录,自定义安装
2.安装CUDA10.0
默认安装目录,自定义安装
3.添加Path
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\lib\x64
检查安装是否成功nvcc -V
(命令行中运行)
4.替换CUDNN
替换至CUDA\v10.0下
5.安装Pytorch
https://download.pytorch.org/whl/torch_stable.html
下载对应版本的torch与torchversion
cu100/torch-1.2.0-cp37-cp37m-win_amd64.whl
cu100/torchvision-0.4.0-cp37-cp37m-win_amd64.whl
(cu100对于cuda10.0,cp37对应python3.7版本)
# 执行安装命令
pip install D:\torch-1.2.0-cp37-cp37m-win_amd64.whl #存放目录随意
pip install D:\torchvision-0.4.0-cp37-cp37m-win_amd64.whl
6.检查GUP是否可用
import torch
torch.cuda.is_available() # True