安裝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)