1、查看cuda版本

原来的老办法是这样的,

cat /usr/local/cuda/version.txt

在我的jetson TX2上的和PC上是一样的,

~$ cat /usr/local/cuda/version.txt
CUDA Version 10.2.300

2、查看cudnn版本

在PC上还是原来的老办法,

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

在我的Jetson TX2上不同,发现在cudnn.h中找不到版本信息,而是在一个叫cudnn_version.h的文件夹里,是这样的,

~$ whereis cudnn_version
cudnn_version: /usr/include/cudnn_version.h

~$ cat /usr/include/cudnn_version.h | grep CUDNN_MAJOR
#define CUDNN_MAJOR 8
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)

~$ cat /usr/include/cudnn_version.h | grep CUDNN_MINOR
#define CUDNN_MINOR 2
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)

所以这里Jetson TX2上的cuda10.2 + cudnn8.2。

本文结束