cuda和tensorflow對應關係
-
查看cuda版本:
cat /usr/local/cuda/version.txt
-
查看cudnn版本:
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
-
查看cuda版本:
nvcc -V
-
查看cudnn版本:
打開cudnn文件夾 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\include ,打開cudnn.h文件,我使用notepad++打開
ctrl+F 檢索 CUDNN_MAJOR ,結果如下:
可判斷我的cudnn版本爲7.1.3