第一步:安装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\binC:\cuda\include\cudnn.h —> C:\Program Files\NVIDIA GPUComputing
Toolkit\CUDA\v9.1\includeC:\cuda\lib\x64\cudnn.lib —> C:\Program Files\NVIDIA GPUComputing
Toolkit\CUDA\v9.1\lib\x64
注意: 一定要安装cudnn,不然无法进行卷积操作
安装GPU版本的pytorch
到官网下载对应GPU版本的pytorch
然后使用pip install 文件名
安装下载好的pytorch压缩包