Win10 安裝 tensorflow-gpu

安裝過程依照 [Win10 安裝 TensorFlow-gpu & Keras]，最後再參考 [Using Virtual Environments in Jupyter Notebook and Python] 將新建的環境加到 Jupyter Notebook 中

作業系統

1. Windows 10 version 1903, OS Build 18362.175
2. 顯卡型號: Nvdia GeForce 940MX

安裝過程

1. 先安裝 python 3.5 的虛擬環境，名稱keras35 (Anaconda prompt@base)

conda create -n keras35 python=3.5
conda activate keras35

2. 為了避免版本衝突，再來一樣是安裝 CUDA 9.0 和 cuDNN 7.0 (08/12更新：cuDNN 7.0，但對應 CUDA 8.0，非9.0)，安裝完後路徑 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin 和 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\libnvvp 就自動加到系統變數 %PATH% 下了

3. 配合 CUDA 的版本，安裝 TensorFlow GPU v1.11 (Anaconda prompt@keras35)

pip install tensorflow-gpu==1.11.0

4. 驗證 TensorFlow 安裝是否成功，執行後應該可以看到 GPU 項目 (Anaconda prompt@keras35 -> python)

from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())

5. 安裝 Keras (Anaconda prompt@keras35)

pip install keras

6. 在 Jupyter Notebook 顯示新環境 (Anaconda prompt@keras35)：先安裝 ipykernel，再將環境加到 Jupyter Notebook。 --name 後接這個新的虛擬環境的名稱，--display-name 是我們想要在 Notebook 中顯示的名稱

pip install --user ipykernel
python -m ipykernel install --user --name keras35 --display-name "Python-keras35"

7. 最後回到 base 環境，啟動 Jupyter Notebook 就可以看到新的 kernel 環境 (Anaconda prompt@base)

ps.後續實作中發現處理 CNN 時還是無法運行(kernel died)，提示 error retrieving driver version: Unimplemented: kernel reported driver version not implemented on Windows，後來參考[此文]修正為「cuDNN 7.0 對應 CUDA 8.0 版本」才成功

<如有轉載，請附上本文連結網址>