最强攻略之CUDA+cudNN安装,tensorflow-gpu,pytorch-gpu一步到位
cuda是做深度学习,后期安装pytorch和anaconda版本必不可少的工具,以下是一些安装步骤,希望对你有用!!
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安装命令到命令行
输入如图命令进行测试
安装成功!!
太牛了!!
欢迎交流!
更多推荐
所有评论(0)