第一次写博客,因为Tensorflow1.4.0(GPU)安装,感慨万千,也希望大家安装的时候别走博主的坑,博主尽量让大家避开坑(关注官网动态很重要)也是看了很多篇博客。但是很多不具有时效性,毕竟Tensorflow一直在发展,一直在更新新版本。总结出了很多问题。
之前在ubuntu14.04下安装Tensorflow0.8.0版本就出了问题,python2.7和3.5选择也出了问题,然后在ubuntu下最后装的是docker。
当然后来发现慢慢支持windows平台后,转移战线,来到了windows平台。
首先,一步一步按照顺序走我的流程,不可以打乱顺序
接下来,我说明我安装的配置情况
这是在一些博客里面看见的,博主本身windows10装过VS所以就不知道没有这个会发生什么。如果你电脑上装了VS,可以跳过这一条。
Anaconda里面集成了很多关于python科学计算的第三方库,主要是安装方便。
在此之前,由于在Windows下TensorFlow目前只支持Python3.5 64-bit版本。
(TensorFlow 从 1.2 开始支持 Python 3.6,之前的官方是不支持的) 所以你可以去Anaconda官方
https://www.anaconda.com/download/
博主安装在F盘,然后等待安装
ps:这张图是网上截的,忽略上面的4.4.0(64bit),我们的是最新版的
到这里,你已经成功安装了Anaconda。
听说安装了Anaconda自带python3.6,博主也不清楚,因为之前博主电脑上就已经预装了3.6,安装完可以尝试在命令行查看python版本。
https://developer.nvidia.com/cuda-gpus
很高兴,博主的游戏本GTX1060是支持GPU的
https://github.com/tensorflow/tensorflow/blob/r1.4/RELEASE.md
博主选了1.4.0版,所以查看它的描述。
显而易见,1.4.0的Tensorflow需要 CUDA8 和 cuDNN6。(这个很重要,看了很多博客众说纷纭,最好去官网查看)
那么我现在先去下载CUDA8,建议下载local版本,在线版本容易出错,安装的时候记住把360之类的软件关了!!!不然容易失败。
这是博主下载的版本。
附上网址(你们真幸福,博主经历的坑都帮你们填补了QAQ)
https://developer.nvidia.com/rdp/form/cudnn-download-survey
别选错了,选错你就gg了,下一步下载Tensorflow就会出错(如图,博主现在一看到红色就紧张)
然后还没完,还要手动添加path,如图(根据自己安装位置添加)
例如博主的是
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\lib\x64
到这里,就算安装完成啦,接下来,该安装Tensorflow了。
对就是这玩意儿,输入:
pip install tensorflow-gpu
(博主之所以没有进度条,是因为之前失败很多次的时候,下载过tensorflow,你们第一次安装,下载完自动安装,会有进度条,耐心等待。)
打开Anaconda Prompt(再次强调,不是cmd),输入python,然后输入以下代码:
import tensorflow as tf
a = tf.random_normal((100, 100))
b = tf.random_normal((100, 500))
c = tf.matmul(a, b)
sess = tf.InteractiveSession()
sess.run(c)
如果有错,欢迎大佬们评论区指正。