Ubuntu20.04安装与使用TensorFlow最新版

我的网站： https://pythoneers.cn

1. pip安装TensorFlow

这里使用的是当前日期最新版的 Ubuntu20.04 作为操作系统平台。20.04默认安装了Python3.8版本，但是没有安装pip，所以可以使用apt先安装pip：

$ sudo apt install python3-pip -y

查看pip的版本，如果不是最新版本顺便将pip升级到最新版本：

$ pip3  -V
$ pip3 install --upgrade pip

安装pip的方式很多，例如可以把pip下载到本地使用Python命令安装等。安装TensorFlow2需要使用高于19.0的pip版本

安装之前需要把pip更新到最新版：

$ pip3 install --upgrade pip

安装 virtualenv 来帮助我们创建虚拟环境，这里使用的是国内阿里云的镜像源，下载速度会快很多：

$ pip3 install -U virtualenv -i https://mirrors.aliyun.com/pypi/simple

创建一个新的 Python 虚拟环境，创建一个 ./venv 目录来存放它：

$ virtualenv ./venv/

如果我们想把系统中已有的库也加入到虚拟环境中，可以加上--system-site-packages。指定Python版本适用-p参数，如：-p python3

激活该虚拟环境：

$ source venv/bin/activate

使用 pip 安装支持 GPU 的 TensorFlow 软件包，可以选择稳定版或预览版。安装稳定版的TensorFlow：

(venv) thanlon@thanlon:~$ pip install tensorflow -i https://mirrors.aliyun.com/pypi/simple/

如果安装只支持CPU版本：pip install tensorflow-cpu；同理安装只支持GPU版本：pip install tensorflow-cpu。指定版本：pip install tensorflow-gpu==2.2.0

安装预览版使用下面的命令：

(venv) thanlon@thanlon:~$ pip install tf-nightly -i https://mirrors.aliyun.com/pypi/simple/

安装之前一定要 保证/tmp空闲足够充足，大概需要3G~4G，当然也可以通过命令临时更改 /tmp 容量：

(venv) thanlon@thanlon:~$