主要是参照官方文档https://www.tensorflow.org/install/install_linux#gpu_support
也参考了网上一些博客,以前都需要安装源码,但是现在不需要了,操作很简单
因为之前机子装了caffe,所以cuda和cudnn都已经是现成的,版本:cuda8.0,cudnn5.0
$sudo apt-get install libcupti-dev
virtualenv安装
$sudo apt-get install python-py python-dev python-virtualenv
$virtualenv --system-site-packages ~/tensorflow
$source ~/tensorflow/bin/activate
$
(tensorflow)$pip install --upgrade tensorflow-gpu #用于Python 2.7和GPU
报错1:
Downloading/unpacking tensorflow-gpu
Could not find any downloads that satisfy the requirement tensorflow-gpu
Cleaning up...
No distributions at all found for tensorflow-gpu
Storing debug log for failure in /tmp/tmpmUPZAM
一般是由于pip版本过低所致,查看了以下pip版本号,
$pip --version
$pip install -U pip
$pip --version
$ pip install --upgrade tensorflow-gpu
看起来似乎是安装成功了
按照官网说明检测一下
报错2:
File "<stdin>", line 1
hello = tf.constant('Hello,TensorFlow!')
^
SyntaxError: invalid syntax
从官网复制的括号变成了中文的括号。。。。
修改了之后
按照成功。撒花~