深度学习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

            参考博客里面已经介绍了下载和版本。这里几个重点。

        深度学习tensorflow --- tensorflow-gpu安装的坑

 

   深度学习tensorflow --- tensorflow-gpu安装的坑

     索性,我直接升级CUDA,升级到最新版,445.

     安装版本:cuda_10.0.130_411.31_windows
下载地址:https://developer.nvidia.com/cuda-10.0-download-archive
深度学习tensorflow --- tensorflow-gpu安装的坑
安装参考:  参考博客2.

深度学习tensorflow --- tensorflow-gpu安装的坑

这里是除了 viaual Studio这个以外全部勾选了

 

cuDNN
下载地址:https://developer.nvidia.com/rdp/cudnn-download (需要注册)
安装版本:cudnn-10.0-windows7-x64-v7.5.0.56
在这里插入图片描述

深度学习tensorflow --- tensorflow-gpu安装的坑

深度学习tensorflow --- tensorflow-gpu安装的坑

深度学习tensorflow --- tensorflow-gpu安装的坑

深度学习tensorflow --- tensorflow-gpu安装的坑

深度学习tensorflow --- tensorflow-gpu安装的坑

 

深度学习tensorflow --- tensorflow-gpu安装的坑

深度学习tensorflow --- tensorflow-gpu安装的坑

安装完成后:在运行-cmd中输入: nvcc –V,显示release 10.0则安装成功。

深度学习tensorflow --- tensorflow-gpu安装的坑

然后:  pip  install  tensorflow-gpu==1.13.1  安装的tensorflow-gpu版本。

 

匹配要点:

1,  CUDA驱动 CUDA toolkit,tensorflow-gpu, python 的匹配关系,前2张图。

2,  安装 CUDA10的时候,不要勾选viaual Studio。

3,配置cudnn。

直接运行,没有问题。

如果有需要留言给我,或者到主页找我联系方式。