我想开始使用 tensorflow-gpu
我看了一些东西,发现我需要确保我有两个。CUDA
和 CUDNN
. 于是,我打开命令提示符,并运行命令 nvidia-smi
检查我 CUDA
版本。
C:\Program Files\NVIDIA Corporation\NVSMI>nvidia-smi
Tue Jun 02 14:13:03 2020
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 445.87 Driver Version: 445.87 CUDA Version: 11.0 |
|-------------------------------+----------------------+----------------------+
| GPU Name TCC/WDDM | Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
|===============================+======================+======================|
| 0 GeForce GTX 1050 WDDM | 00000000:01:00.0 Off | N/A |
| N/A 40C P8 N/A / N/A | 77MiB / 4096MiB | 0% Default |
+-------------------------------+----------------------+----------------------+
+-----------------------------------------------------------------------------+
| Processes: |
| GPU PID Type Process name GPU Memory |
| Usage |
|=============================================================================|
| 0 10488 C+G ...n64\EpicGamesLauncher.exe N/A |
| 0 12636 C+G ...4\UnrealCEFSubProcess.exe N/A |
+-----------------------------------------------------------------------------+
现在我看到我的 CUDA
版本是 11.0
我去了 NVidia网站 选择一个版本的 CUDNN
能与 CUDA 11.0
但最新的支持到 CUDA 10.2
目前。我应该怎么做?我可以用那个 CUDA 10.2
?
什么 nvidia-smi
显示的不是你安装的CUDA版本,而是你的驱动所支持的最大CUDA版本。
CUDA 11.0已经公布但尚未发布(截至2020年6月2日),所以你应该使用CUDA 10.2,因为它是最新的版本。