深度学习入门(一)安装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)
3、选择合适的tensorFlow-GPU 版本,比如1.5-1.12支持CUDA9和Cudnn7。(很重要:本文最开始安装的1.12.0却提示不成功(不明白原因),后来只好降到1.10.0)
二:安装步骤
---------------------------------------------------------------------------------------------------------
安装Cudnn
由于之前已安装CUDA9.0,这里就不说怎么安装了(网上一大堆)。
所以直接把Cudnn的三个文件夹拷贝到CUDA环境下合并就可以了,其实就三个文件。
同时配置下环境变量(不配置后面启用tensorflow-gpu会报错)
---------------------------------------------------------------------------------------------------------
---------------------------------------------------------------------------------------------------------
1、安装Anaconda3,比较简单,一步到底
安装完后打开Anaconda Prompt可查看信息
2、安装python3.6。在Anaconda Prompt用命令()即可快速安装
GPU版本:conda create --name tensorflow-gpu python=3.6
CPU版本:conda create --name tensorflow python=3.6
安装完成后**一个新环境:然后就可以看到除了base外新添加了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系统
直接从官网安装,速度太慢了,而且连接容易断掉,
所以用大家喜欢的清华镜像(https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu==1.11.0 后面指定你要安装的版本),哈哈。(这里截图是1.11.0是安装失败时候截得图,后面安装的是1.10.0)
安装完成后输入 import tensorflow as tf查看是否安装成功
三、测试程序。在Anaconda中Spyder下测试
ok,终于安装成功,Peace and Love
参考一下博客:(比较有用)
I:https://blog.****.net/gangeqian2/article/details/79358543 (写得简单明了)
II:https://blog.****.net/sunmingyang1987/article/details/102872658