初学者进化（1）：pytorch的cuda环境搭建（GPU版本安装）全攻略

#**

初学者进化： 用gpu跑pytorch全过程攻略

在训练神经网络这方面，gpu的效率能达到cpu的几十倍，所以想要学好神经网络，gpu的配置是必要的。
本文仅限于pytorch环境的gpu安装（不过cuda和cudnn环境是通用的）

作者也是一点点bug改过来的，深感找文章麻烦，所以我把自己的心路历程与遇到的bug写下这篇文章，以期对后入有益，便足矣。

我总结的安装要求大概是对应版本的cuda+cudnn+pytorch+torchvision+python环境，每一项都要和版本一一对应。（作者还遇到了新建环境后spyder打不开的情况，会在附录中加上）

1.cuda安装

cuda的安装教程十分简单，参考下述教程：
https://blog.csdn.net/qq_36653505/article/details/81368346

感谢大佬无私帮助

2.cudnn安装

cudnn主要用于卷积等运算，是pytorch学习模型的依赖
下载对应版本的cudnn链接：https://developer.nvidia.com/rdp/cudnn-archive
这里要根据下载的cuda版本，寻找对应的cudnn

打开压缩包，依次将压缩包中的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.1\lib\x64

ps：本文为CSDN博主「Never-Giveup」的原创文章，在这引用，感谢大佬的无私帮助。原文链接：https://blog.csdn.net/qq_36653505/article/details/83932941

cuda和cudnn都需要配置对应的环境变量：