1. 程式人生 > >Windows10 64位安裝TensorFlow-

Windows10 64位安裝TensorFlow-

安裝前準備: 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

來安裝1.5版本。