目錄
Linux
查看 CUDA 版本
- 方法一：
nvcc --version
或
nvcc -V
如果 nvcc 沒有安裝，那么用方法二。
- 方法二：
去安裝目錄下查看：
cat /usr/local/cuda/version.txt
查看 cuDNN 版本
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
如果沒有，那么可能沒有安裝 cuDNN。
如果是為了使用 PyTorch/TensorFlow，在 Linux 服務器上推薦使用 conda 安裝，使用 conda 可以很方便安裝 PyTorch/TensorFlow 以及對應版本的 CUDA 和 cuDNN。
Windows
查看 CUDA 版本
在命令行中執行：
nvcc --version
或者進入 CUDA 的安裝目錄查看：
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA
查看 cuDNN 版本
進入 CUDA 的安裝目錄查看文件 cudnn.h ：（注意修改v9.0）
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\include\cudnn.h
如下所示，cuDNN 版本為 7.2.1 :
使用 PyTorch 查看 CUDA 和 cuDNN 版本
import torch
print(torch.__version__)
print(torch.version.cuda)
print(torch.backends.cudnn.version())
References
ubuntu: 查看cuda版本 -- JNingWei
Windows查看CUDA版本 -- 胡大炮的妖孽人生
LINUX下查詢CUDA和CUDNN版本 -- 許文_北京北