Windows安装TensorFlow
為什么80%的碼農(nóng)都做不了架構(gòu)師?>>> ??
本文將解釋如何在Windows下安裝TensorFlow。
確定安裝哪類TensorFlow
需要先確定哪種類型的TensorFlow:
- 僅支持CUP運算版本:如果電腦的系統(tǒng)沒有?NVIDIA?的GPU,那么必須安裝這個版本。這個版本的TensorFlow安裝非常簡單(安裝僅需一個命令,5到10分鐘),所以即使系統(tǒng)中有滿足要求的NVIDIA? GPU官方還是建議在學(xué)習(xí)階段安裝這個版本。
- 支持GPU運算的版本:TensorFlow程序在GPU下運行比在CPU下運行明顯快很多。如果系統(tǒng)中包含?NVIDIA?的GPU滿足下一個小節(jié)所示的條件并且程序?qū)π阅芤蠛芨?#xff0c;建議安裝此版本。
運行TensorFlow所需要的GPU配置
如果在系統(tǒng)中安裝使用GPU運行的TensorFlow,需要確保下面介紹的NVIDIA軟件已經(jīng)安裝到系統(tǒng)中。
- CUDA? Toolkit 8.0。請看 NVIDIA安裝cuda 的文檔,根據(jù)文檔中的描述確保已經(jīng)將CUDA相關(guān)的路徑增加到?%PATH%?環(huán)境變量中。
- NVIDIA的驅(qū)動關(guān)聯(lián)?CUDA Toolkit 8.0。
- cuDNN v5.1。請查看 NVIDIA?cudnn 文檔。需要注意的是cuDNN通常安裝在與其他CUDA動態(tài)鏈接庫(dll)不同的位置。確保已經(jīng)將cuDNN的 動態(tài)鏈接庫(dll)的地址添加到系統(tǒng)的??%PATH%?環(huán)境變量中。
- GPU顯卡必須擁有3.0以上版本的CUDA計算能力,查看 NVIDIA顯卡支持列表 了解支持情況。
如果系統(tǒng)中已經(jīng)安裝了以前的相關(guān)包,請更新到所指定的版本。
如何安裝TensorFlow
在安裝TensorFlow之前必須選定一個安裝機制。目前提供2種機制:
- "native"app
- Anaconda
Native的安裝(以下簡稱本地安裝)方式會將TensorFlow直接安裝在當前的系統(tǒng)中,不會在系統(tǒng)和TensorFlow之間搭建任何的虛擬環(huán)境,所以本地安裝不會額外安裝一個獨立的容器。需要注意的是本地安裝可能會干擾系統(tǒng)中其他基于python安裝的程序。如果事先已經(jīng)安裝配置了滿足需要的python環(huán)境,本地安裝通常只需要一個命令就可以完成。使用本地安裝,用戶可以在系統(tǒng)中任何位置運行TensorFlow。
在Anaconda模式下,需要使用conda創(chuàng)建一個虛擬環(huán)境。官方優(yōu)先推薦使用?pip install?命令來安裝TensorFlow,其次再考慮anaconda的?conda install?命令。conda包是第三方社區(qū)提供的(非TensorFlow官方),TensorFlow團隊從始至終都不會去測試在conda中運行的情況,在使用時需考慮這個風(fēng)險。
本地安裝
首先,需要安裝以下版本的python:
- Python 3.5.x from python.org
TensorFlow在windows操作系統(tǒng)中僅僅支持3.5.x版本的python。Python 3.5.x附帶pip3軟件包管理器,這是用于安裝TensorFlow的程序。
安裝TensorFlow需要啟動一個終端(terminal),然后在該終端中輸入對應(yīng)的pip3 install命令。安裝僅支持CPU版本的TensorFlow,輸入以下命令:
C:\> pip3 install --upgrade tensorflow安裝GPU版本的TensorFlow,使用以下命令:
C:\> pip3 install --upgrade tensorflow-gpuAnaconda模式安裝
再次強調(diào),Anaconda安裝是有第三方社區(qū)提供的,非官方。
在Anaconda環(huán)境中安裝TensorFlow分為以下幾個步驟:
?
?鍵入以下命令來啟用conda環(huán)境:
C:> activate tensorflow(tensorflow)C:> # Your prompt should change?
鍵入以下命令在conda環(huán)境中安裝TensorFlow。這里 安裝CPU版本的命令:
(tensorflow)C:> pip install --ignore-installed --upgrade https://storage.googleapis.com/tensorflow/windows/cpu/tensorflow-1.0.1-cp35-cp35m-win_amd64.whl這是GPU版本的命令:
(tensorflow)C:> pip install --ignore-installed --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-1.0.1-cp35-cp35m-win_amd64.whl?
驗證安裝?
如果python輸出以下內(nèi)容,則表明TensorFlow已經(jīng)安裝成功然后就可以寫TensorFlow的程序了:
如果收到了一些異常信息,請繼續(xù)向下看。
常見的安裝問題
TensorFlow通過Stack Overflow網(wǎng)站來記錄錯誤信息以及處理方法。下面的列表包含一些跳轉(zhuǎn)的到?Stack Overflow的連接。如果在安裝過程中遇到的問題沒有在下面中,請到Stack Overflow去搜索相關(guān)的關(guān)鍵字。若還是搜索不到,請直接提出新問題并標記?tensorflow?的標簽。
| 41007279 | ? [...\stream_executor\dso_loader.cc] Couldn't open CUDA library nvcuda.dll |
| 41007279 | ? [...\stream_executor\cuda\cuda_dnn.cc] Unable to load cuDNN DSO |
| 42006320 | ? ImportError: Traceback (most recent call last): File "...\tensorflow\core\framework\graph_pb2.py", line 6, in from google.protobuf import descriptor as _descriptor ImportError: cannot import name 'descriptor' |
| 42011070 | ? No module named "pywrap_tensorflow" |
轉(zhuǎn)載于:https://my.oschina.net/chkui/blog/881871
總結(jié)
以上是生活随笔為你收集整理的Windows安装TensorFlow的全部內(nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: JS window对象的top、pare
- 下一篇: 正则表达式选项分析