ubuntu16.04安装TensorFlow

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

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

图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版本,都要更新一下,具体命令如下:

 
  1. python 2.7版本:sudo pip install --upgrade pip

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

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

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

 
  1. [global]

  2. index-url = https://pypi.tuna.tsinghua.edu.cn/simple

  3. [install]

  4. trusted-host=mirrors.aliyun.com

五、安装TensorFlow

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

 
  1. python 2.7版本:sudo pip install tensorflow

  2. python 3.x版本:sudo pip3 install tensorflow


GPU版本安装命令如下:

 
  1. python 2.7版本:sudo pip install tensorflow-gpu

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


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

六、测试安装结果

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

图2

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/y1250056491/article/details/78670710

猜你喜欢

转载自blog.csdn.net/taotongning/article/details/81288855
今日推荐