Linux中安装TensorFlow教程-Ubuntu18.04安装TensorFlow教程-一步一步完全版!

一、所需工具

Ubuntu系统(版本16.04及以上,例如18.04)

二、说明

本教程采用pip安装TensorFlow,而且采用谷歌推荐的Create a virtual environment (recommended)方法安装,使得包安装与系统隔离,防止产生混乱

三、安装步骤

1、首先检查是否满足了所需的环境,输入下面三个命令:

python3 --version
pip3 --version
virtualenv --version

2 、如果版本号正确输出,表示都安好了,直接看第4步,否则继续往下看第3步。
3、如果没安好,使用下面的命令将python3、pip3和virtualenv安好。

sudo apt update
sudo apt install python3-dev python3-pip
sudo pip3 install -U virtualenv  # system-wide install

4、创建虚拟环境。

virtualenv --system-site-packages -p python3 ./venv

注意, ./venv表示的是你的虚拟环境的位置,请改成你想创建的位置。(这里我们创建在当前目录的venv文件夹下)
5、激活虚拟环境。

source ./venv/bin/activate

如果你前面创建虚拟环境时改了路径,这里也要改成:你的虚拟环境的路径/bin/activate
此时,你的shell前面都会以(venv)开头了。
6、更新pip。注意,由于我们是在虚拟环境里操作,这些操作都不会影响系统设置,都是在虚拟环境里进行的。

pip install --upgrade pip
pip list  # show packages installed within the virtual environment

7、安装TensorFlow。以CPU版本为例,使用下面的命令。

pip install --upgrade tensorflow

如果你想安装其他版本,根据下面的说明选一个就行了:

Choose one of the following TensorFlow packages to install from PyPI:
tensorflow —Latest stable release for CPU-only (recommended for beginners)
tensorflow-gpu —Latest stable release with GPU support (Ubuntu and Windows)
tf-nightly —Preview build (unstable). Ubuntu and Windows include GPU support.

8、安装完成后,测试安装结果。输入下面的命令:

python -c "import tensorflow as tf;print(tf.reduce_sum(tf.random.normal([1000, 1000])))"

如果输出的最后一行是tf.tensor什么什么,恭喜你安装完成!
9、使用下面的命令退出虚拟环境。

deactivate

注意,因为我们是在虚拟环境里安装的TensorFlow,所以,你必须在使用TensorFlow前进入虚拟环境,在用完TensorFlow后再退出虚拟环境否则TensorFlow是无法使用的。所以请仔细阅读下面的文字

四、注意事项

因为我们是在虚拟环境里安装的TensorFlow,所以,你必须在使用TensorFlow前进入虚拟环境,在用完TensorFlow后再退出虚拟环境否则TensorFlow是无法使用的。
进入虚拟环境的命令:

source ./venv/bin/activate

如果你前面创建虚拟环境时改了路径,这里也要改成:你的虚拟环境的路径/bin/activate
进入虚拟环境后会,shell会变成类似在这里插入图片描述开头
退出虚拟环境的命令:

deactivate