- 查看cuda版本
cat /usr/local/cuda/version.txt
或者
nvcc --version
或者 nvcc -V
- 查看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 */