cuda 版本检查:
nvcc -V
cat /usr/local/cuda/version.txt
cudnn 版本检查:
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
或者(不同版本命令不同)
cat /usr/include/cudnn.h | grep CUDNN_MAJOR -A 2
参考:https://medium.com/@changrongko/nv-how-to-check-cuda-and-cudnn-version-e05aa21daf6c