文档首页/ AI开发平台ModelArts/ 常见问题/ Standard训练作业/ 功能咨询/ 如何查询自定义镜像的cuda和cudnn版本?
更新时间:2024-10-28 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