pytorch的cuda环境搭建（GPU版本安装）

第一步：安装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个文件复制到相应位置

1. C:\cuda\bin\cudnn64_7.dll —> C:\Program Files\NVIDIA GPUComputing
   Toolkit\CUDA\v9.1\bin

2. C:\cuda\include\cudnn.h —> C:\Program Files\NVIDIA GPUComputing
   Toolkit\CUDA\v9.1\include

3. C:\cuda\lib\x64\cudnn.lib —> C:\Program Files\NVIDIA GPUComputing
   Toolkit\CUDA\v9.1\lib\x64

注意： 一定要安装cudnn，不然无法进行卷积操作

安装GPU版本的pytorch

到官网下载对应GPU版本的pytorch

然后使用pip install 文件名安装下载好的pytorch压缩包

第四步：检查是否安装成功

参考

1. https://blog.csdn.net/MarsLee_U/article/details/80489618
2. https://blog.csdn.net/u011622208/article/details/80275344