1. 在系统上安装 Python 开发环境
需要 Python 3.4、3.5 或 3.6
python3 --version
pip3 --version
virtualenv --version
若没有安装:sudo apt update
sudo apt install python3-dev python3-pip
sudo pip3 install -U virtualenv # system-wide install
2. 创建虚拟环境(推荐)
Python 虚拟环境用于将软件包安装与系统隔离开来。
创建一个新的虚拟环境,方法是选择 Python 解析器并创建一个 ./venv
目录来存放它:
virtualenv --system-site-packages -p python3 ./venv
使用特定于 shell 的命令激活该虚拟环境:
source ./venv/bin/activate # sh, bash, ksh, or zsh
当 virtualenv 处于有效状态时,shell 提示符带有 (venv)
前缀。
在不影响主机系统设置的情况下,在虚拟环境中安装软件包。首先升级 pip
:
pip install --upgrade pip
pip list # show packages installed within the virtual environment
之后要退出 virtualenv,请使用以下命令:
deactivate # don't exit until you're done using TensorFlow
3. 安装 TensorFlow pip 软件包
请从 PyPI 中选择以下某个 TensorFlow 软件包进行安装:
tensorflow
- 仅支持 CPU 的最新稳定版(建议新手使用)tensorflow-gpu
- 支持 GPU 的最新稳定版(适用于 Ubuntu 和 Windows)tf-nightly
- 仅支持 CPU 的预览每夜版(不稳定)tf-nightly-gpu
- 支持 GPU 的预览每夜版(不稳定,适用于 Ubuntu 和 Windows)tensorflow==2.0.0-alpha0
- 仅支持 CPU 的预览 TF 2.0 Alpha 版(不稳定)tensorflow-gpu==2.0.0-alpha0
- 支持 GPU 的预览 TF 2.0 Alpha 版(不稳定,Ubuntu 和 Windows)
进入虚拟环境:
source ./venv/bin/activate
然后,安装tf:
# Current stable release for CPU-only pip install tensorflow
验证安装效果:
python -c "import tensorflow as tf; tf.enable_eager_execution(); print(tf.reduce_sum(tf.random_normal([1000, 1000])))"