windows系统:
cuda:
nvcc -V
或者打开cuda安装文件夹 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0 中的 version.txt 文件查看
我的cuda版本为9.0.176
也可以直接 win+R 打开命令行cmd, 输入以下命令进行查询
cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v
type version.txt
cudnn:
打开cudnn文件夹 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\include
打开cudnn.h文件,我使用notepad++打开
ctrl+F 检索 CUDNN_MAJOR ,结果如下
根据上面结果判断我的cudnn版本为7.0.3
win+R 打开命令行cmd, cd到cudnn文件夹
cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\include
查看cudnn是否安装成功:
cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\extras\demo_suite
在命令行打开分别打开bandwidthTest.exe
和deviceQuery.exe
两个Result = PASS则成功
linux系统:
cuda:
nvcc -V
或者
cat /usr/local/cuda/version.txt
cudnn:
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2