集群管理——(5)ubuntu下安装tensorflow(gpu)

来源:互联网 发布:线路板画图软件 编辑:程序博客网 时间:2024/06/05 13:30

之前已经安装了cuda8 cudnn6.0,caffe,现在配置一下tensorflow.
安装依赖:

$sudo apt-get install libcupti-dev
$sudo apt-get install python-py python-dev python-virtualenv

按照官方文档的virtualenv方式安装:
注意安装位置,习惯把软件放在/usr/local

$virtualenv --system-site-packages /usr/local/tensorflow
$source /usr/local/tensorflow/bin/activate

执行成功后变成 (tensorflow) $

$(tensorflow)$pip install --upgrade tensorflow-gpu  #python2.7+gpu

可能会报错:

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:
$ sudo -m pip install -U pip

升级以后问题就能得到解决。

再次执行安装命令:

$(tensorflow)$pip install --upgrade tensorflow-gpu 

等待下载安装,完成后就能在python里面调用tensorflow 了。