Windows10 64位安裝TensorFlow-
阿新 • • 發佈:2018-12-20
安裝前準備:
1.顯示卡支援CUDA
2.python3.5以上。64位(TensorFlow1.2 開始支援 Python 3.6,之前的官方是不支援的)注意:python一定要是64位的,否則安裝不上!
3.確保有穩定的網路連線
4.確保pip版本>=8.1(用 pip -V
檢視當前 pip 版本,用 python -m pip install -U pip
升級pip )
5.建議安裝 Anaconda,因為這個集成了很多科學計算所必需的庫,能夠避免很多依賴問題
如果滿足上述條件,開始安裝。 1.開始下載 CUDA 安裝包 點選進入下載
選擇適合你的版本。
像安裝其他軟體一樣,正常安裝。 安裝過程:
輸入nvcc -V
檢視版本資訊。如圖表示安裝成功。
2.安裝cuDNN
點選下載
需要註冊。然後才能下載。我下載的是這個版本。
下載後是一個壓縮包,進行解壓。解壓放到任何一個目錄下就行,然後把其中的bin目錄路徑新增到Path環境變數裡。 這個是我的路徑。
在命令提示符中使用pip install tensorflow-gpu
安裝tensorflow-gpu
安裝結果:
編寫程式進行測試遇到了這個問題:
新版本的 TensorFlow會出現此問題,根據 TensorFlow 的釋出說明,該版本會使用 AVX 指令,所以可能會在舊 CPU 上不能執行:
使用舊版 TensorFlow,比如你可以使用 pip install tensorflow-gpu=1.5