如果之前下载了别的torch或是pytorch最好先删除,不然后面安装torch总有奇奇怪怪的问题。
一、
cuda+cudnn先前已经安装好的。选择和显卡对应版本的cuda下载安装即可。
再选对应cudnn下载
打开压缩包,依次将压缩包中的3个文件复制到相应位置
- …\cuda\bin\cudnn64_7.dll —> …\NVIDIA GPUComputing
Toolkit\CUDA\v9.1\bin - …\cuda\include\cudnn.h —> …\NVIDIA GPUComputing
Toolkit\CUDA\v9.1\include - …\cuda\lib\x64\cudnn.lib —>…\NVIDIA GPUComputing
Toolkit\CUDA\v9.1\lib\x64
二、pytorch
根据python版本和cuda版本去官网https://pytorch.org/下载对应whl文件.(这是我下载的版本win+python37+cuda9.2+torch0.4.1老版本的下载入口)
在whl文件下载目录下cmd进入命令行,pip install安装(前面加的是国内镜像源)
pip install -i https://pypi.doubanio.com/simple/ --trusted-host pypi.doubanio.com torch-0.4.1-cp37-cp37m-win_amd64.whl
三、验证安装是否成功
import torch
print(torch.__version__)
print(torch.version.cuda)
print(torch.cuda.is_available())