Tensorflow-GPU环境安装-Windows版

Tensorflow-GPU=CUDA-10.0+ANACODA-3.0+CUDNN-7.6.4

  1. 进入Anaconda的官网下载,选择自己想下载的平台版本+Python版本,最后点击Download按钮就OK了。下载完成后接下来就是傻瓜式安装,一路确认,想改路径改一下路径就可以。详细图文教程点我
  2. 安装好ANACODA后需要确认自己的机子需要什么版本的CUDA和CUDNN,然后在命令行窗口输入命令  

         pip install tensorflow            

         pip install tensorflow-gpu

         安装tensorflow和tensorflow-gpu

    3.安装成功后,去tensorflow的目录下去找D:\software\ANACONDA\Lib\site-packages\tensorflow_core\python\platform
      找到一个build_info.py文件 查看自己需要的CUDA和CUDNN版本

Tensorflow-GPU环境安装-Windows版

    4.去CUDA官网下载CUDA软件,笔者是CUDA10.0  

点击 Legacy Releases查看历史版本:

Tensorflow-GPU环境安装-Windows版

 

选择自己需要的版本:

Tensorflow-GPU环境安装-Windows版

  然后选择windows版本后下载就完事了。Tensorflow-GPU环境安装-Windows版

 

 

5.傻瓜式安装,一路同意点下去。

    6.之后去CUDNN官网下载与CUDA相匹配的CUDNN版本。(需要注册一下)我的cuda10.0对应cudnn 7.6.4。

Tensorflow-GPU环境安装-Windows版
    7.下载完成后将CUDNN解压缩并拷贝CUDNN中的三个文件夹到CUDA文件的路径下。

Tensorflow-GPU环境安装-Windows版

Tensorflow-GPU环境安装-Windows版


    8.此时还需要将这三个文件的路径添加到环境变量中去。

Tensorflow-GPU环境安装-Windows版

最后可以测试一下自己是否安装成功:

import tensorflow as tf

print(tf.test.is_gpu_available())

输出是True就OK啦!

 

Tensorflow-GPU环境安装-Windows版

恭喜你可以开始自己的深度学习从入门到放弃的路程啦。、

作者微信:

Tensorflow-GPU环境安装-Windows版

相关文章: