深度学习入门(一)安装TensorFlow的GPU版本

 

安装软件:VS2015    CUDA9.0  Anaconda3  TensorFlow-GPU 1.10.0

一:注意问题

1、选择合适的Python版本,工具Anaconda3自带的Python版本是3.7.4,(网上说的好多包没有3.7的,还是选择3.6比较稳妥)

2、根据CUDA版本确定Cudnn的版本(我下载的库版本是cudnn-9.0-windows10-x64-v7.6.4.38)

深度学习入门(一)安装TensorFlow的GPU版本

3、选择合适的tensorFlow-GPU 版本,比如1.5-1.12支持CUDA9和Cudnn7。(很重要:本文最开始安装的1.12.0却提示不成功(不明白原因),后来只好降到1.10.0)

深度学习入门(一)安装TensorFlow的GPU版本

 

 

二:安装步骤

---------------------------------------------------------------------------------------------------------

安装Cudnn

由于之前已安装CUDA9.0,这里就不说怎么安装了(网上一大堆)。

所以直接把Cudnn的三个文件夹拷贝到CUDA环境下合并就可以了,其实就三个文件。

同时配置下环境变量(不配置后面启用tensorflow-gpu会报错)

---------------------------------------------------------------------------------------------------------

深度学习入门(一)安装TensorFlow的GPU版本

深度学习入门(一)安装TensorFlow的GPU版本

---------------------------------------------------------------------------------------------------------

1、安装Anaconda3,比较简单,一步到底

深度学习入门(一)安装TensorFlow的GPU版本

安装完后打开Anaconda Prompt可查看信息

深度学习入门(一)安装TensorFlow的GPU版本

 

2、安装python3.6。在Anaconda Prompt用命令()即可快速安装

GPU版本:conda create --name tensorflow-gpu python=3.6 

CPU版本:conda create --name tensorflow python=3.6 

深度学习入门(一)安装TensorFlow的GPU版本

安装完成后**一个新环境:然后就可以看到除了base外新添加了tensorflow-gpu

深度学习入门(一)安装TensorFlow的GPU版本

3、安装tenserflow-gpu ,安装前可查看(命令:anaconda search -t conda tensorflow-gpu)支持的tenserflow-gpu版本

tensorflow与pyhon版本以及操作系统有兼容性

例:tensorflow_gpu-2.0.0a0-cp37-cp37m-win_amd64.whl

gpu-2.0版本,cp37代表python3.7版本,win代表windows系统

深度学习入门(一)安装TensorFlow的GPU版本

 

 

直接从官网安装,速度太慢了,而且连接容易断掉,

所以用大家喜欢的清华镜像(https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu==1.11.0  后面指定你要安装的版本),哈哈。(这里截图是1.11.0是安装失败时候截得图,后面安装的是1.10.0)

深度学习入门(一)安装TensorFlow的GPU版本

 

 

深度学习入门(一)安装TensorFlow的GPU版本

安装完成后输入 import tensorflow as tf查看是否安装成功

深度学习入门(一)安装TensorFlow的GPU版本

 

三、测试程序。在Anaconda中Spyder下测试

深度学习入门(一)安装TensorFlow的GPU版本

ok,终于安装成功,Peace and Love

参考一下博客:(比较有用)

I:https://blog.****.net/gangeqian2/article/details/79358543  (写得简单明了)

II:https://blog.****.net/sunmingyang1987/article/details/102872658