深度学习tensorflow --- tensorflow-gpu安装的坑
如果百度的话,你会发现很多的安装文献,大体相同,但是都有些不一样的地方。我的安装过程和百度上的也不太一样。所以,在这里也提供一个方法给安装tensorflow-gpu的同志们参考。
环境 Python3.7, gpu是mx150 , tensorflow-gpu 是1.13.1 , win10 操作系统 , cuda 10 , driverVersion 445, cuDnn 7.4
查看你电脑的cuda版本:https://jingyan.baidu.com/article/d169e1861b759f436611d882.html
参考博客1:https://blog.csdn.net/weixin_44095128/article/details/88664928
参考博客2:https://blog.csdn.net/qq_39295665/article/details/90581702
参考博客里面已经介绍了下载和版本。这里几个重点。
索性,我直接升级CUDA,升级到最新版,445.
安装版本:cuda_10.0.130_411.31_windows
下载地址:https://developer.nvidia.com/cuda-10.0-download-archive
安装参考: 参考博客2.
这里是除了 viaual Studio这个以外全部勾选了。
cuDNN
下载地址:https://developer.nvidia.com/rdp/cudnn-download (需要注册)
安装版本:cudnn-10.0-windows7-x64-v7.5.0.56
在这里插入图片描述
安装完成后:在运行-cmd中输入: nvcc –V,显示release 10.0则安装成功。
然后: pip install tensorflow-gpu==1.13.1 安装的tensorflow-gpu版本。
匹配要点:
1, CUDA驱动 CUDA toolkit,tensorflow-gpu, python 的匹配关系,前2张图。
2, 安装 CUDA10的时候,不要勾选viaual Studio。
3,配置cudnn。
直接运行,没有问题。
如果有需要留言给我,或者到主页找我联系方式。