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安装命令到命令行

输入如图命令进行测试

安装成功!!

太牛了!!

欢迎交流!

Logo

欢迎加入我们的广州开发者社区,与优秀的开发者共同成长!

更多推荐