安裝cuDNN
安裝完CUDA之後,還需下載對應的cuDNN版本後並完成相關設定,才能正式開始使用CUDA GPU運算。
下載cuDNN
1.需先註冊並登入會員
2.登入會員後下載 cuDNN
搭配組合(持續更新)
| tensorflow-gpu | CUDA | cuDNN | 更新時間 |
|---|---|---|---|
| 2.1 | 10.1 | >= 7.6 (for CUDA 10.1) | 2020.2 |
| 2.0 | 10.0/10.1 | >= 7.4.1 (for CUDA 10.0/10.1) | 2019.10 |
| 1.13 | 10.0 | >= 7.4.1 (for CUDA 10.0) | 2019.3 |
| 1.12 | 9.0 | >= 7.2 (for CUDA 9.0) | 2018.11 |
| 1.11 | 9.0 | >= 7.2 (for CUDA 9.0) | 2018.11 |
| 1.10 | 9.0 | 7.1.4 (for CUDA 9.0) | 2018.9 |
| 1.9 | 9.0 | 7.1.4 | 2018.9 |
| 1.8 | 9.0 | 7.1.4 | 2018.7 |
| 1.7 | 9.0 | 7.1.2 | 2018.4 |
| 1.5 | 9.0 | 7(.0.4/5) | 2018.2 |
| 1.3 | 8.0 | 6.0 | 2017.11 |
| 1.0 | 8.0 | 5.1 | 2017.8 |
(官方說明:https://www.tensorflow.org/install/gpu)
下載非最新版cuDNN:https://developer.nvidia.com/rdp/cudnn-archive
3.開啟下載檔後將cuda裡的三個資料夾(bin, include, lib) 複製並取代原資料夾
目標路徑:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0

4.設定系統環境變數

選擇Path,點選編輯

