TensorFlow 2 軟件包現(xiàn)已推出
- tensorflow:支持 CPU 和 GPU 的最新穩(wěn)定版(適用于 Ubuntu 和 Windows)
- tf-nightly:預覽 build(不穩(wěn)定)禀挫。Ubuntu 和 Windows 均包含 GPU 支持。
舊版 TensorFlow
對于 TensorFlow 1.x,CPU 和 GPU 軟件包是分開的:
- tensorflow==1.15:僅支持 CPU 的版本
- tensorflow-gpu==1.15:支持 GPU 的版本(適用于 Ubuntu 和 Windows)
系統(tǒng)要求
- Python 3.5-3.7
- pip 19.0 或更高版本(需要 manylinux2010 支持)
- Ubuntu 16.04 或更高版本(64 位)
- macOS 10.12.6 (Sierra) 或更高版本(64 位)(不支持 GPU)
- Windows 7 或更高版本(64 位)(僅支持 Python 3)
- 適用于 Visual Studio 2015晶疼、2017 和 2019 的 Microsoft Visual C++ 可再發(fā)行軟件包
- Raspbian 9.0 或更高版本
- GPU 支持需要使用支持 CUDA? 的顯卡(適用于 Ubuntu 和 Windows)
注意:必須使用最新版本的 pip,才能安裝 TensorFlow 2。
硬件要求
- 從 TensorFlow 1.6 開始,二進制文件使用 AVX 指令昌阿,這些指令可能無法在舊版 CPU 上運行。
- 閱讀 GPU 支持指南恳邀,以在 Ubuntu 或 Windows 上設置支持 CUDA? 的 GPU 卡懦冰。
1. 在系統(tǒng)上安裝 Python 開發(fā)環(huán)境(Python 3)
檢查是否已配置 Python 環(huán)境:
需要使用 Python 3.5-3.7 和 pip 19.0 及更高版本
python3 --version
pip3 --version
virtualenv --version
如果已安裝這些軟件包,請?zhí)料乱徊健?br> 否則谣沸,請安裝 Python刷钢、pip 軟件包管理器和 Virtualenv:
Ubuntu
sudo apt update
sudo apt install python3-dev python3-pip
sudo pip3 install -U virtualenv # system-wide install
macOS
使用 Homebrew 軟件包管理器安裝:
/usr/bin/ruby -e "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/master/install)"
export PATH="/usr/local/bin:/usr/local/sbin:$PATH"
brew update
brew install python # Python 3
sudo pip3 install -U virtualenv # system-wide install
注意:升級系統(tǒng) pip 可能會導致問題。
如果不是在虛擬環(huán)境中乳附,請針對下面的命令使用 python3 -m pip内地。這樣可以確保您升級并使用 Python pip,而不是系統(tǒng) pip许溅。
2.創(chuàng)建虛擬環(huán)境(推薦)
Python 虛擬環(huán)境用于將軟件包安裝與系統(tǒng)隔離開來瓤鼻。
Ubuntu/macOS
創(chuàng)建一個新的虛擬環(huán)境,方法是選擇 Python 解釋器并創(chuàng)建一個 ./venv
目錄來存放它:
virtualenv --system-site-packages -p python3 ./venv
使用特定于 shell 的命令激活該虛擬環(huán)境:
source ./venv/bin/activate # sh, bash, ksh, or zsh
當 virtualenv
處于有效狀態(tài)時贤重,shell 提示符帶有 (venv)
前綴。
在不影響主機系統(tǒng)設置的情況下清焕,在虛擬環(huán)境中安裝軟件包并蝗。首先升級 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 和 GPU 的最新穩(wěn)定版(適用于 Ubuntu 和 Windows)。
- tf-nightly:預覽 build(不穩(wěn)定)秸妥。Ubuntu 和 Windows 均包含 GPU 支持滚停。
- tensorflow==1.15:TensorFlow 1.x 的最終版本。
系統(tǒng)會自動安裝軟件包依賴項粥惧。這些依賴項就列在 setup.py 文件的 REQUIRED_PACKAGES 下键畴。
pip install --upgrade tensorflow
驗證安裝效果:
python -c "import tensorflow as tf;print(tf.reduce_sum(tf.random.normal([1000, 1000])))"
成功:TensorFlow 現(xiàn)已安裝完畢。請查看教程開始使用突雪。
軟件包位置
部分安裝方式需要您提供 TensorFlow Python 軟件包的網(wǎng)址起惕。您需要根據(jù) Python 版本指定網(wǎng)址。
參考
https://www.tensorflow.org/install/pip?lang=python2#ubuntu-mac-os