文档首页> AI开发平台ModelArts> 常见问题> 训练作业> 功能咨询> 如何查询自定义镜像的cuda和cudnn版本?
更新时间:2023-01-09 GMT+08:00

如何查询自定义镜像的cuda和cudnn版本?

查询cuda版本:

cat /usr/local/cuda/version.txt

查询cudnn版本:

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