如何在Ubuntu 18.04上安装TensorFlow
TensorFlow是Google建立的一个免费的开源机器学习平台。 Twitter,PayPal,英特尔,联想和空中客车等许多组织都使用它。
TensorFlow可以在系统范围内,Python虚拟环境中,作为Docker容器或与Anaconda一起安装。 出于学习目的,最好在Python虚拟环境中安装TensorFlow。 这样,您可以在一台计算机上拥有多个不同的隔离Python环境,并在每个项目的基础上安装模块的特定版本,而不必担心会影响您的其他Project。
本教程介绍了如何在Ubuntu 18.04上安装TensorFlow。
在Ubuntu 18.04上安装TensorFlow#
以下各节提供了有关如何在Ubuntu 18.04上的Python虚拟环境中安装TensorFlow的逐步说明。
1.安装Python 3和venv#
默认情况下,Ubuntu 18.04随附Python 3.6。 您可以通过键入以下命令来验证系统上已安装Python 3:
python3 -V
输出应如下所示:
Python 3.6.6
如果要改用Python 3.8,请查看这些指南。
从Python 3.6开始,建议的创建虚拟环境的方法是使用 venv
模块。 要安装 python3-venv
提供以下内容的软件包 venv
模块运行以下命令:
sudo apt install python3-venv
安装模块后,我们准备为TensorFlow项目创建虚拟环境。
2.创建虚拟环境#
首先,导航到要存储Python 3虚拟环境的目录。 它可以是您的主目录,也可以是用户具有读写权限的任何其他目录。
为TensorFlow项目创建一个新目录并cd到其中:
mkdir my_tensorflow
cd my_tensorflow
进入目录后,运行以下命令来创建虚拟环境:
python3 -m venv venv
上面的命令创建一个名为 venv
,其中包含Python二进制文件,Pip程序包管理器,标准Python库和其他支持文件的副本。 您可以为虚拟环境使用任何名称。
要开始使用此虚拟环境,您需要通过运行 activate
脚本:
source venv/bin/activate
激活后,虚拟环境的bin目录将被添加到 $PATH
可变的。 而且,shell的提示符也会更改,并且会显示您当前正在使用的虚拟环境的名称。 在这种情况下 venv
。
TensorFlow安装要求 pip
版本19或更高版本。 运行以下命令进行升级 pip
到最新版本:
pip install --upgrade pip
3.安装TensorFlow#
现在已经激活了虚拟环境,是时候安装TensorFlow软件包了。
pip install --upgrade tensorflow
如果您拥有专用的NVIDIA GPU,并希望利用其处理能力,而不是 tensorflow
安装 tensorflow-gpu
包含GPU支持的软件包。
在虚拟环境中,可以使用以下命令 pip
代替 pip3
和 python
代替 python3
。
要验证安装,请使用以下命令来打印TensorFlow版本:
python -c 'import tensorflow as tf; print(tf.__version__)'
在撰写本文时,TensorFlow的最新稳定版本是2.0.0。
2.0.0
您的TensorFlow版本可能与此处显示的版本不同。
如果您不熟悉TensorFlow,请访问TensorFlow入门页面并了解如何构建第一个ML应用程序。 您还可以从Github克隆TensorFlow模型或TensorFlow-Examples存储库,并探索和测试TensorFlow示例。
完成工作后,通过键入以下内容停用环境 deactivate
然后您将返回到常规外壳。
deactivate
结论#
在本教程中,我们向您展示了如何在Ubuntu 18.04上安装TensorFlow。
如果您遇到问题或有反馈,请在下面发表评论。
python ubuntu