1. 程式人生 > >windows 8 + Tensorflow 1.10.0 + Python 3.6.4 + CUDA 9.0 + CUDNN7.3.0配置

windows 8 + Tensorflow 1.10.0 + Python 3.6.4 + CUDA 9.0 + CUDNN7.3.0配置

windows 8 + Tensorflow 1.10.0 + Python 3.6.4 + CUDA 9.0 + CUDNN7.3.0配置

一、首先安裝TensorFlow

因為anaconda可以更方便的進行包的管理,所以先裝了一波anaconda,然後pip install tensorflow-gpu

二. CUDA 9.0安裝(注意CUDA9.1和CUDA9.2還不行,親測)

(1)安裝CUDA
到如下網址下載CUDA 9.0
https://developer.nvidia.com/cuda-toolkit-archive
CUDA 9.0下載
(2) 驗證CUDA安裝是否成功:
開啟命令提示符,輸入:nvcc -V
出現如下類似資訊:
安裝成功

三. CUDNN7.3.0下載

選擇對應的版本下載,解壓縮,將資料夾裡的內容拷貝到CUDA的安裝目錄並覆蓋相應的資料夾,CUDA預設安裝目錄:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0
CUDNN

執行時可能的報錯

Internal: cudaGetDevice() failed. Status: CUDA driver version is insufficient for CUDA runtime version

解決方法:更新驅動
drivers驅動下載
https://www.geforce.com/drivers

其他命令

檢查當前GPU執行狀態
cd C:\Program Files\NVIDIA Corporation\NVSMI\
nvidia-smi.exe