Tensorflow gpu win10環境配置
阿新 • • 發佈:2018-12-16
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)初始化例程失敗。