使用 pip 安裝 TensorFlow

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)址。

版本 網(wǎng)址
Linux
Python 2.7(支持 GPU) https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-2.1.0-cp27-cp27mu-manylinux2010_x86_64.whl
Python 2.7(僅支持 CPU) https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow_cpu-2.1.0-cp27-cp27mu-manylinux2010_x86_64.whl
Python 3.5(支持 GPU) https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-2.1.0-cp35-cp35m-manylinux2010_x86_64.whl
Python 3.5(僅支持 CPU) https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow_cpu-2.1.0-cp35-cp35m-manylinux2010_x86_64.whl
Python 3.6(支持 GPU) https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-2.1.0-cp36-cp36m-manylinux2010_x86_64.whl
Python 3.6(僅支持 CPU) https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow_cpu-2.1.0-cp36-cp36m-manylinux2010_x86_64.whl
Python 3.7(支持 GPU) https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-2.1.0-cp37-cp37m-manylinux2010_x86_64.whl
Python 3.7(僅支持 CPU) https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow_cpu-2.1.0-cp37-cp37m-manylinux2010_x86_64.whl
macOS(僅支持 CPU)
Python 2.7 https://storage.googleapis.com/tensorflow/mac/cpu/tensorflow-2.1.0-cp27-cp27m-macosx_10_9_x86_64.whl
Python 3.5 https://storage.googleapis.com/tensorflow/mac/cpu/tensorflow-2.1.0-cp35-cp35m-macosx_10_6_intel.whl
Python 3.6 https://storage.googleapis.com/tensorflow/mac/cpu/tensorflow-2.1.0-cp36-cp36m-macosx_10_9_x86_64.whl
Python 3.7 https://storage.googleapis.com/tensorflow/mac/cpu/tensorflow-2.1.0-cp37-cp37m-macosx_10_9_x86_64.whl
Windows
Python 3.5(支持 GPU) https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-2.1.0-cp35-cp35m-win_amd64.whl
Python 3.5(僅支持 CPU) https://storage.googleapis.com/tensorflow/windows/cpu/tensorflow_cpu-2.1.0-cp35-cp35m-win_amd64.whl
Python 3.6(支持 GPU) https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-2.1.0-cp36-cp36m-win_amd64.whl
Python 3.6(僅支持 CPU) https://storage.googleapis.com/tensorflow/windows/cpu/tensorflow_cpu-2.1.0-cp36-cp36m-win_amd64.whl
Python 3.7(支持 GPU) https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-2.1.0-cp37-cp37m-win_amd64.whl
Python 3.7(僅支持 CPU) https://storage.googleapis.com/tensorflow/windows/cpu/tensorflow_cpu-2.1.0-cp37-cp37m-win_amd64.whl
Raspberry PI(僅支持 CPU)
Python 3咏删、Pi0 或 Pi1 https://storage.googleapis.com/tensorflow/raspberrypi/tensorflow-2.1.0-cp35-none-linux_armv6l.whl
Python 3惹想、Pi2 或 Pi3 https://storage.googleapis.com/tensorflow/raspberrypi/tensorflow-2.1.0-cp35-none-linux_armv7l.whl

參考

https://www.tensorflow.org/install/pip?lang=python2#ubuntu-mac-os

?著作權歸作者所有,轉載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市督函,隨后出現(xiàn)的幾起案子嘀粱,更是在濱河造成了極大的恐慌激挪,老刑警劉巖,帶你破解...
    沈念sama閱讀 222,729評論 6 517
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件锋叨,死亡現(xiàn)場離奇詭異垄分,居然都是意外死亡,警方通過查閱死者的電腦和手機娃磺,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 95,226評論 3 399
  • 文/潘曉璐 我一進店門薄湿,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人豌鸡,你說我怎么就攤上這事嘿般。” “怎么了涯冠?”我有些...
    開封第一講書人閱讀 169,461評論 0 362
  • 文/不壞的土叔 我叫張陵炉奴,是天一觀的道長。 經(jīng)常有香客問我蛇更,道長瞻赶,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 60,135評論 1 300
  • 正文 為了忘掉前任派任,我火速辦了婚禮砸逊,結果婚禮上,老公的妹妹穿的比我還像新娘掌逛。我一直安慰自己师逸,他們只是感情好,可當我...
    茶點故事閱讀 69,130評論 6 398
  • 文/花漫 我一把揭開白布豆混。 她就那樣靜靜地躺著篓像,像睡著了一般。 火紅的嫁衣襯著肌膚如雪皿伺。 梳的紋絲不亂的頭發(fā)上员辩,一...
    開封第一講書人閱讀 52,736評論 1 312
  • 那天,我揣著相機與錄音鸵鸥,去河邊找鬼奠滑。 笑死,一個胖子當著我的面吹牛妒穴,可吹牛的內(nèi)容都是我干的宋税。 我是一名探鬼主播,決...
    沈念sama閱讀 41,179評論 3 422
  • 文/蒼蘭香墨 我猛地睜開眼宰翅,長吁一口氣:“原來是場噩夢啊……” “哼弃甥!你這毒婦竟也來了?” 一聲冷哼從身側響起汁讼,我...
    開封第一講書人閱讀 40,124評論 0 277
  • 序言:老撾萬榮一對情侶失蹤淆攻,失蹤者是張志新(化名)和其女友劉穎阔墩,沒想到半個月后,有當?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體瓶珊,經(jīng)...
    沈念sama閱讀 46,657評論 1 320
  • 正文 獨居荒郊野嶺守林人離奇死亡啸箫,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 38,723評論 3 342
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發(fā)現(xiàn)自己被綠了伞芹。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片忘苛。...
    茶點故事閱讀 40,872評論 1 353
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖唱较,靈堂內(nèi)的尸體忽然破棺而出扎唾,到底是詐尸還是另有隱情,我是刑警寧澤南缓,帶...
    沈念sama閱讀 36,533評論 5 351
  • 正文 年R本政府宣布胸遇,位于F島的核電站,受9級特大地震影響汉形,放射性物質(zhì)發(fā)生泄漏纸镊。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點故事閱讀 42,213評論 3 336
  • 文/蒙蒙 一概疆、第九天 我趴在偏房一處隱蔽的房頂上張望逗威。 院中可真熱鬧,春花似錦岔冀、人聲如沸凯旭。這莊子的主人今日做“春日...
    開封第一講書人閱讀 32,700評論 0 25
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽尽纽。三九已至,卻和暖如春童漩,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背春锋。 一陣腳步聲響...
    開封第一講書人閱讀 33,819評論 1 274
  • 我被黑心中介騙來泰國打工矫膨, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人期奔。 一個月前我還...
    沈念sama閱讀 49,304評論 3 379
  • 正文 我出身青樓侧馅,卻偏偏與公主長得像,于是被迫代替她去往敵國和親呐萌。 傳聞我的和親對象是個殘疾皇子馁痴,可洞房花燭夜當晚...
    茶點故事閱讀 45,876評論 2 361