如何检测cuda和cudnn的版本?

内容目录

check cuda

cat /usr/local/cuda/version.txt

nvcc --version

check cudnn:

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

or

cat /usr/include/cudnn.h | grep CUDNN_MAJOR -A 2

打赏作者