深度学习--记录深度学习环境搭配(win10+GTX1060+CUDA9.0+cuDNN7.4+tensorflow-gpu1.2)

写在前面:网上关于初步接触深度学习并搭配环境的博客有许多,本文记录博主初步学习深度学习并搭配环境的过程

https://blog.csdn.net/hejunqing14/article/details/76059603

https://blog.csdn.net/qq_33856151/article/details/79295086

https://blog.csdn.net/qq_37296487/article/details/83028394

 

初始环境:win10英文系统、python3.6

第一步、CUDA安装

a,查看显卡版本:Control Panel —> NVIDIA控制面板 —> help —>System Information

深度学习--记录深度学习环境搭配(win10+GTX1060+CUDA9.0+cuDNN7.4+tensorflow-gpu1.2)_第1张图片

b,版本匹配,官网地址:https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html#cuda-developer-tools-new-features,也可以参考这篇博客:https://blog.csdn.net/omodao1/article/details/83241074

深度学习--记录深度学习环境搭配(win10+GTX1060+CUDA9.0+cuDNN7.4+tensorflow-gpu1.2)_第2张图片

c,选择对应的CUDA版本进行安装,官网下载地址:https://developer.nvidia.com/cuda-toolkit-archive。自定义安装,取消勾选Visual Studio Integration,然后一路安装,中间也可以返回。

d,测试是否安装成功,cmd命令行:NVCC -V

到此,CUDA安装成功,完毕。

第二步、cuDNN安装

a,官网下载,下载地址:https://developer.nvidia.com/rdp/cudnn-download

里面有对应CUDA版本的cuDNN版本,下载对应的cuDNN,友情提示,需要注册才能下载!

b,解压后配置到CUDA目录里面,然后添加环境变量,参考博客:https://blog.csdn.net/qq_37296487/article/details/83028394

第三步、tensorflow-gpu安装

pip install tensorflow-gpu

 

接下来,小白继续加油!

你可能感兴趣的:(深度学习)