Tensorflow安装教程

环境

  • GTX1080Ti
  • Ubuntu14.04LTS(阿里的源)

步骤

  1. 安装CUDA8.0
  2. 安装Cudnn
  3. 安装pip
  4. 安装Tensorflow

安装CUDA8.0

总体上的思路参考这篇文章(ubuntu14.04下tensorflow安装指南(GPU)),但是其中坑很多。

  • 下载好安装文件,注意,这个安装文件官方有一版本是错误的,现在已经修复。文件名都是cuda_8.0.61_375.26_linux.run,window下是1.4G大小那个是对的。否则直接安装不了。
  • 安装时,最好不要选择安装opengl,会和ubuntu冲突不能再次打开图形界面了!!!
  • 安装完成后,可能想切换成图形界面,碰到登陆循环的问题,解决方案网上一大堆,但是都没去试。最后我是root下卸载了nvidia的驱动重启就直接成功了。具体原因因为一开始装ubuntu后没有显卡驱动,字都很大模糊,然后先装了驱动。
nvidia-uninstall

坑主要三个,就够受的了。后来问题解决了,找到一篇文章,不错,作者也是用心解决的,不像那些随便抄抄抄的。ubuntu16.04+cuda8.0+cudnn5.1+caffe 安装(基于独立主机&集成显卡&GTX1080Ti)

安装Cudnn

这个好像没什么太多的坑,照着第一个文章走没有什么问题。唯一不确定的是在/usr/local/目录下有两个,一个cuda还有一个cuda-8.0,最后我选择两个都复制了一遍。

安装pip

本身ubuntu就有python2和3,Tensorflow好像基于2的。

sudo apt-get install python-pip
sudo apt-get install python-dev

安装Tensorflow

安装Tensorflow其实很简单,被网上很多的弄得很麻烦,安装网上的一些版本他们都去寻找cuda7.0或者7.5的,所以还是让他自己下载安装吧。对了,网上的那种还要翻墙。

sudo apt-get install tensorflow

测试

>>> import tensorflow as tf
>>> hello = tf.constant('Hello, TensorFlow!')
>>> sess = tf.Session()
>>> print sess.run(hello)
Hello, TensorFlow!
>>> a = tf.constant(10)
>>> b = tf.constant(32)
>>> print sess.run(a+b)
42
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

友情链接更多精彩内容