Ubuntu20.04确认cuda和cudnn已经安装成功

当我们通过官网安装cuda和cudnn时，终端执行完命令后我们仍不能确定是否已经安装成功。接下来教大家用几句命令测试。

cuda

检测版本号

nvcc -V

如果输出如下，则安装成功。

可以看到版本号是11.2

cudnn

检测版本号

有两种命令：如果你的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

正确输出如图