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

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

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

2.安装pip:

wget https://bootstrap.pypa.io/get-pip.py 
sudo python get-pip.py

3.升级pip:

    $ sudo -m pip install -U pip  

4.搭建沙盒环境

$virtualenv --system-site-packages /usr/local/tensorflow

一:在沙盒中安装tensorflow

$source /usr/local/tensorflow/bin/activate

执行成功后变成 (tensorflow) $

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

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

二:直接安装:(推荐,当然也要在四个前提下)
1.安装仅支持cpu版本的

pip install tensorflow==1.1.0 

2.安装支持gpu版本的,前提是安装了CUDA SDK,(这里我已经安装了CUDA8.0)

pip install tensorflow-gpu==1.1.0 

你可能感兴趣的:(ubuntu-配置,集群)