我爱编程

Tensorflow安装教程

2017-12-05  本文已影响0人  Mr丶sorrow

环境

步骤

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

安装CUDA8.0

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

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
上一篇 下一篇

猜你喜欢

热点阅读