Ubuntu安装TensorFlow GPU教程介绍TensorFlow是一个开源机器学习平台。谷歌设计该软件是为了帮助研究人员、数据科学家和开发人员简化实现机器学习模型的过程。
Ubuntu如何安装TensorFlow GPU?这个用于数值计算的端到端库可以在多个 CPU、GPU 以及移动操作系统上运行。在本教程中,了解如何在 Ubuntu 18.04 上安装到 TensorFlow。
文章图片
先决条件
- Ubuntu Linux 系统(16.04 或更高版本)
- 访问命令行/终端窗口 ( Ctrl + Alt + T )
- 具有sudo 权限的用户帐户
- 点 19.0 或更高版本
- Python(3.4 或更高版本)
- 在PIP包管理器(不超过19.0版本以上)
- Virtualenv(用于隔离 Python 环境的软件)
sudo apt update
接下来,使用以下命令安装前两个软件包:
sudo apt install python3-dev python3-pip
文章图片
【如何在Ubuntu 18.04上安装TensorFlow GPU(分步指南)】然后,为 Virtualenv 的广泛系统安装运行以下命令:
sudo pip3 install -U virtualenv
文章图片
在新的./venv目录中创建你的第一个环境:
virtualenv --system-site-packages -p python3 ./venv
文章图片
然后,激活虚拟环境以开始在其中工作。运行以下命令:
source ./venv/bin/activate
你的 shell 提示现在应该有一个(venv)前缀,如下图所示:
文章图片
激活venv 后,继续在隔离环境中安装pip:
pip install --upgrade pip
文章图片
如果要查看虚拟环境中所有包的完整列表,请使用以下命令:
pip list
它显示所有软件包及其各自的版本,如下图所示:
文章图片
选项 1:为 CPU 安装 TensorFlow
默认的TensorFlow软件包支持基于 CPU 的工作负载。要安装包及其依赖项,请键入以下命令:
pip install --upgrade tensorflow
选项 2:为 GPU 安装 TensorFlow
Ubuntu安装TensorFlow GPU教程:TensorFlow for GPU需要专用的 NVIDIA CUDA?-enabled GPU 和相关驱动程序。该软件包适用于基于 GPU 的机器学习工作负载。
要为 GPU安装最新的稳定版 TensorFlow ,请运行以下命令:
pip install --upgrade tensorflow-gpu
有关 TensorFlow for GPU 的硬件和软件要求列表,请参阅TensorFlow 关于 GPU 支持的文档。
选项 3:安装旧版本的 TensorFlow
用于 CPU 和 GPU 的旧版TensorFlow也可供下载。
如何在Ubuntu安装TensorFlow GPU?通过以以下格式运行命令来安装1.14 及更早版本:
pip install tensorflow==package_version
要为 CPU 1.14安装TensorFlow,请运行以下命令:
pip install tensorflow==1.14
要为 GPU 1.14安装TensorFlow,请运行以下命令:
pip install tensorflow-gpu==1.14
1.15 版支持基于 CPU 和 GPU 的工作负载。要安装TensorFlow 1.15,请输入以下命令:
pip install tensorflow-gpu==1.15rc2
第 4 步:验证 TensorFlow 安装要在 Ubuntu 中验证 TensorFlow 的安装,请在终端窗口中输入命令:
python -c "import tensorflow as tf;
print(tf.reduce_sum(tf.random.normal([
1000, 1000])))"
Ubuntu安装TensorFlow GPU教程结论Ubuntu如何安装TensorFlow GPU?本文向你展示了在 Ubuntu 18.04 上安装 TensorFlow 的步骤。系统上安装了所需的TensorFlow 版本(CPU 或 GPU 支持)后,你现在可以继续开发机器学习模型。
推荐阅读
- 22个用于编程和编码的最佳Linux文本编辑器合集
- WIN10专业版激活工具图文图文详细教程
- win10变回win7图文图文详细教程
- 最新通用版win10激活密钥分享自制步骤
- win10 64位 产品密钥分享自制步骤
- windows10专业版激活密钥分享自制步骤
- 安装系统 极速一键重装系统图文图文详细教程
- windows10 企业版 激活图文图文详细教程
- 最容易的xp安装双系统办法