本文的前提是已经安装好了navida显卡驱动、CUDA8.0、cudnn5.1具体安装过程可以参考本人往期博客与他人博客。
Tensorflow有很多种安装方式,这里只介绍一种我用的pip安装方式。
首先要选定自己要安装的tensorflow版本
到这个网站https://github.com/tensorflow/tensorflow
根据自己情况,是否有gpu支持、python的不同版本。点击对应的build history。
进去后在左侧栏选择自己需要安装的版本,tensorflow各个版本不尽相同,建议根据自己需要选择,这里版主选的是0.12.1。链接为https://ci.tensorflow.org/view/Nightly/job/nightly-matrix-linux-gpu/TF_BUILD_IS_OPT=OPT,TF_BUILD_IS_PIP=PIP,TF_BUILD_PYTHON_VERSION=PYTHON2,label=gpu-linux/388/
选择一个项目右键复制链接地址。
在终端里sudo pip install --upgrade https://ci.tensorflow.org/view/Nightly/job/nightly-matrix-linux-gpu/TF_BUILD_IS_OPT=OPT,TF_BUILD_IS_PIP=PIP,TF_BUILD_PYTHON_VERSION=PYTHON2,label=gpu-linux/388/artifact/pip_test/whl/tensorflow_gpu-0.12.1-cp27-none-linux_x86_64.whl。
upgrade后面的链接即为你刚刚右键复制的链接。
然后就安装成功啦!
就这么简单!
最后,检验一下安装是否成功。
~$ python
>>> import tensorflow as tf
>>> hello = tf.constant('Hello,Tensor')
>>> sess=tf.Session()
>>> print(sess.run(hello))
如果可以显示Hello,Tensor,就说明没问题了!
再试一个
>>> a=tf.constant(10)
>>> b=tf.constant(22)
>>> print sess.run(a+b)
出现32,说明没问题。
如果中途报错,那说明安装有问题,请仔细复习哪里出错。
祝大家都可以顺利安装成功!