安裝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,點選編輯

將以下路徑新增至PATH中 (請根據CUDA版本名稱自行調整路徑)

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin;
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\extras\CUPTI\libx64;
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\include;

results matching ""

    No results matching ""