ubuntu安装TensorFlow
在安装TensorFlow前,有四个主要选项:
- Python 2.x
- Python 2.x + GPU
- Python 3.x
- Python 3.x + GPU
一般ubuntu系统自带Python2.X,但是我使用的版本是Python3.X,所以需要安装Python3以及pip3。
本贴只介绍了不带GPU版本的Python3.X的安装方法,其他版本安装见下文参考博客
安装Python3:
sudo apt install python3
同理:pip也要对应安装的python版本,三代对应:python3-pip
安装pip:
sudo apt-get install python3-pip python3-dev
然后把pip更新到最新版本
sudo pip3 install --upgrade pip
验证
最后查看版本来检查是否安装好相应的版本
python3 -V
pip3 -V
安装TensorFlow
注意:一定要版本对应,因为我用的是pip3,所以通过pip3 install
来下载国内镜像源,直接下载官网的文件巨慢。
sudo pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple/ https://mirror.tuna.tsinghua.edu.cn/tensorflow/linux/cpu/tensorflow-1.7.0-cp35-cp35m-linux_x86_64.whl
TensorFlow验证
打开Python解释器,在命令行输入
python3
注意,输入python
默认打开python2.X。
然后导入TensorFlow库:
import tensorflow as tf
请注意,如果没有正确安装TensorFlow,或者将带GPU的版本安装在不受支持的系统上,就会在此处出现错误。CUDA错误在这一步上非常普遍。如果有效,就可以试试打印出TensorFlow的版本:
print(tf.__version__)
(注意version前后有2个下划线)
若打印出版本号即安装成功。
对于Python2.X版本的安装或者带GPU的版本的安装参考以下几篇文章。
参考文章: