ubuntu16.04安装TensorFlow的正确步骤

一、查看系统所安装的python版本

打开终端输入指令:python,如图1所示,我的系统是ubuntu16.04.03,默认安装的python版本为2.7.12。

ubuntu16.04安装TensorFlow的正确步骤

图1

二、安装python对应版本的pip和依赖包

若python版本为2.7,则输入如下命令:

sudo apt-get install python-pip python-dev

若python版本为3.x,则输入如下命令:

sudo apt-get install python3-pip python3-dev

三、升级pip版本

在装tensorflow之前,不管是不是最新的pip版本,都要更新一下,具体命令如下:

python 2.7版本:sudo pip install --upgrade pip
python 3.x版本:sudo pip3 install --upgrade pip

四、更改pip源地址(提高下载速度)

修改 ~/.pip/pip.conf (没有就创建一个文件夹及文件,文件夹要加".",表示是隐藏文件夹),内容如下:

[global]
index-url = https://pypi.tuna.tsinghua.edu.cn/simple
[install]
trusted-host=mirrors.aliyun.com

五、安装TensorFlow

TensorFlow可以安装CPU和GPU两种版本,CPU版本安装命令如下:

python 2.7版本:sudo pip install tensorflow
python 3.x版本:sudo pip3 install tensorflow

GPU版本安装命令如下:

python 2.7版本:sudo pip install tensorflow-gpu
python 3.x版本:sudo pip3 install tensorflow-gpu

若上述命令执行过程没有报错,则安装成功。

六、测试安装结果

进入python编译环境,导入TensorFlow,做一个简单的加法运算,如图2所示。

ubuntu16.04安装TensorFlow的正确步骤

ubuntu16.04安装TensorFlow的正确步骤

相关推荐