英伟达官方文档

  1. 查看cuda版本

cat /usr/local/cuda/version.txt
或者
nvcc --version 或者 nvcc -V

  1. 查看cudnn版本

网上都是这个但是不行
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
后来改为这个成功了
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

$ cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
#define CUDNN_MAJOR 8
#define CUDNN_MINOR 2
#define CUDNN_PATCHLEVEL 0
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)

#endif /* CUDNN_VERSION_H */

Logo

更多推荐