第一步:安装cuda
安装cuda,这里还需要安装对应版本的Visual Studio(具体参考本博主的博客 https://blog.csdn.net/qq_36653505/article/details/81368346 )
第二步:安装cudnn
cudnn是pytorch搭建深度学习模型的依赖,没有它,不能运行卷积等操作。
下载对应版本的cudnn链接:https://developer.nvidia.com/rdp/cudnn-archive
打开压缩包,依次将压缩包中的3个文件复制到相应位置
-
C:\cuda\bin\cudnn64_7.dll —> C:\Program Files\NVIDIA GPUComputing
Toolkit\CUDA\v9.1\bin -
C:\cuda\include\cudnn.h —> C:\Program Files\NVIDIA GPUComputing
Toolkit\CUDA\v9.1\include -
C:\cuda\lib\x64\cudnn.lib —> C:\Program Files\NVIDIA GPUComputing
Toolkit\CUDA\v9.