1. 程式人生 > >Tensorflow gpu win10環境配置

Tensorflow gpu win10環境配置

Tensorflow gpu win10

1.安裝cuda 9.0 2.解壓cudnn-9.0-windows10-x64-v7.1.zip 複製解壓檔案目錄下 bin,include,lib 三個資料夾到 cuda根目錄 (預設在C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0)

3.安裝Anaconda 3 4.開啟Anaconda Prompt,建立環境

conda create -n tf-gpu python=3.6

5.進入環境

conda activate tf-gpu

6.升級pip到最新版,防止稍後的安裝時,出現錯誤

python -
m pip install --upgrade pip

7.安裝tensorflow-gpu版本 (注:奔騰系列處理器可能需要低版本,tensorflow==1.5,原因:不支援AVX指令集

pip install tensorflow-gpu ==1.11.0

8.驗證:在命令列中,進入python,並輸入以下程式碼:

import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))

輸出如下,即為環境配置成功

b’ Hello, TensorFlow

注:1.如果import tensorflow時出現以下的錯誤,可以檢查下處理器是否為奔騰系列,是否支援AVX指令集 換成低版本的tensorflow即可 (如1.5)

ImportError: DLL load failed: 動態連結庫(DLL)初始化例程失敗。