最强攻略之CUDA+cudNN安装,tensorflow-gpu，pytorch-gpu一步到位

cuda是做深度学习，后期安装pytorch和anaconda版本必不可少的工具，以下是一些安装步骤，希望对你有用！！

第一步，打开NVIDIA 控制面板

右下角系统信息，打开，点击“组件”查看最高支持的CUDA版本

cuda版本决定了，你后期安装pytorch-gpu和tensorflow-gpu版本的问题，最后选择一个主流版本

鄙人安装了CUDA10.1 教程也是关于CUDA10.1的

tensorflow-gpu 对应的CUDA

连接：在 Windows 环境中从源代码构建  |  TensorFlow

第二步，下载和安装CUDA

链接：CUDA Toolkit 12.2 Update 2 Downloads | NVIDIA Developer

或者百度搜索CUDA10.1,如图所示，下载即可

下载exe文件之后，双击即可

“同意并继续”

选“自定义”

选择如上，下一步

下一步

下一步

完成之后在“高级系统设置中”->"环境变量"Path中，增加cuda两个路径，如图确认即可

在命令行输入

nvcc -V

显示如下就是成功了

第三步，安装cudnn

cuda10.1对应的cudnn为v7.6.5

链接：CUDA Deep Neural Network (cuDNN) | NVIDIA Developer
https://developer.nvidia.com/rdp/cudnn-archive

cudnn需要注册账号才可以下载

下载之后是个压缩包文件,解压之后，把文件夹下的三个文件复制到cuda路径之下，如图

大功告成

第四步，tensorflow-gpu安装

tensorflow安装比较容易，主要看上面安装的什么版本

在命令行中输入

pip install tensorflow-gpu==2.1.0(版本号)

测试一下

安装成功，非常完美！！

第五步，pytorch-gpu安装

pytorch官网：Previous PyTorch Versions | PyTorch

找到cuda10.1 对应版本，复制pip安装命令到命令行

输入如图命令进行测试

安装成功！！

太牛了！！

欢迎交流！