集群管理——(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 
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值