Ubuntu18.04安装tensorflow-gpu(简单 快捷)

本文介绍2种安装方法:

方法1:
开发环境:python3
1.安装Tensorflow命令:

pip3 install tensorflow

测试是否安装成功:

python3
import tensorflow

2.安装Anaconda(目的是为了将tensorflow环境并入)
参考:Anaconda安装 选择Aanconda3
(1)安装Anaconda命令:wget https://repo.continuum.io/archive/Anaconda3-5.1.0-Linux-x86_64.sh

(2) 运行下载好的Anaconda,找到下载的目录,然后执行命令:bash XXXXXXXXX(就是Anaconda文件的名字)

(3)当执行完成上面的步骤之后,对Anaconda 的环境进行测试;

conda --version

若提示不能找到版本,则:
修改环境变量

gedit ~/.bashrc

在最后添加:

export PATH=~/anaconda3/bin:$PATH

然后再执行

conda --version

(4)添加tensorflow的环境。执行命令:

conda create -n tensorflow python=3.6

(版本的话,我个人比较喜欢3.X+版本)。当执行完成之后,就根据提示,进行输入yes就可以了,慢慢等待。

(5)**环境,执行命令:(作用:进入到tensorflow的环境)

source activate tensorflow

(6)进行测试:在tensorflow的环境下,执行命令:python (作用:表示进入python环境)

(7)然后输入代码(这个其实和windows安装的时候测试时一样):

import tensorflow as tf  
hello = tf.constant('first tensorflow')  
sess = tf.Session()  
print (sess.run(hello))

如果:输出first tensorflow ,那么就表示安装成功了。
补充内容:

1:当需要退出python环境,即执行Ctrl+D或者输入quit即可
2:退出tensorflow环境,source deactivate
3:**tensorflow环境,source activate tensorflow
参考:Ubuntu18.04+Tensorflow 环境配置
方法2:
官方推荐通过Virtualenv来安装tensorflow-gpu,来进行项目隔离,这是在开发环境,如果在生产环境,可以走docker方式来部署。这里采用Virtualenv方式来安装,执行如下命令:

sudo apt-get install python3
sudo apt-get install python3-pip python3-dev python-virtualenv

安装完pip3、virtualenv后,可进入virtualenv安装tensorflow-gpu,执行如下命令:

mkdir -p ~/tensorflow/venv
virtualenv -p python3 ~/tensorflow/venv #创建virtualenv环境
cd ~/tensorflow/venv
source bin/activate #环境用的时候需要**
pip3 install --index-url http://pypi.douban.com/simple --trusted-host pypi.douban.com --upgrade tensorflow-gpu #采用国内源下载tensorflow-gpu

最后通过 pip3 show tensorflow-gpu 可检查tensorflow-gpu的安装路径是否在该虚拟环境内,以隔绝与全局的site-packages
检查结果
  还是在上述virtualenv创建的根目录下,执行如下命令:

mkdir src && cd src
vi hello_tensorflow.py

vi hello_tensorflow.py,内容如下:

import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))

最后执行该脚本:

python3 hello_tensorflow.py

即可得到包含有GPU输出信息的正确结果    Ubuntu18.04安装tensorflow-gpu(简单 快捷)注意:使用的时候需要**

cd ~/tensorflow/venv
source bin/activate #环境用的时候需要**

参考:
深度学习应用系列(一)| 在Ubuntu 18.04安装tensorflow 1.10 GPU版本

查看安装版本和路径:

 import tensorflow as tf
tf.__version__  # version ID
tf.__path__      # installation path

Ubuntu18.04安装tensorflow-gpu(简单 快捷)