網站:NVIDIA Driver Downloads
請到他們的網站選擇自己的版本,下載後安裝
網站:NVIDIA cuDNN
請到他們的網站,中間有個 Download,接著要加入會員登入
填寫完基本需求調查後,點選 Process to download
打勾 I Agree ...
務必選擇此版本 Download cuDNN v6.0 (April 27, 2017), for CUDA 8.0
並且挑選作業系統 cuDNN v6.0 Library for Windows 10
下載好後會是一個資料夾中包含三個子資料夾 bin/include/lib
將 bin 的路徑加到系統環境變數(path)中
網站:CUDA GPUs
要先確定自己的GPU符不符合資格,點選上述網站後
點開 CUDA-Enabled GeForce Products
找到自己的 GPU 名稱,務必大於3.0
這個數值代表他們具備的功能,而非效率
1080跟1050都是6.1,但運算速度還是有差
網站:CUDA Toolkit 8.0 - Feb 2017
確定好後,到上述網站下載 CUDA 8.0
選擇好環境後,Installer Type 使用 local
檔案蠻大的,下載好後安裝即可
過程中若出現,未偵測到對應的GPU,別理它繼續安裝
因為像筆電就有內險(GPU 0)跟外顯(GPU 1)
你的外顯符合資格,但它只會比對內險
安裝好後,確認有將
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\libnvvp
加到環境變數,通常會自動加入,若無再手動加入即可
開啟 Anaconda Prompt 依序輸入以下指令,安裝 tensorflow-gpu
這時候才可以使用這條指令
(直接使用,會安裝不能互相對應的版本,才需要以上步驟先安裝)
開啟 Anaconda Prompt 依序輸入以下指令