如何在Google Vertex AI运行环境中更新nvidia-driver版本

AI 小北

2026-10-1

Vertex AI自定义容器NVIDIA驱动版本升级方案

核心结论

无法直接在运行的自定义容器内修改主机节点预装的NVIDIA驱动，可通过以下几种可行方案满足驱动版本≥455的要求：

方案1：选用Vertex AI官方提供的训练CUDA基础镜像构建自定义容器，选择CUDA 11.2及以上版本的镜像即可。Vertex AI调度作业时会自动分配预装了符合版本要求驱动的主机节点，无需额外配置即可直接使用。
方案2：如果必须使用自研基础镜像，将镜像内部的CUDA工具包版本升级到11.2及以上，提交训练作业时添加环境变量 NVIDIA_DISABLE_REQUIREMENTS=1 关闭默认的驱动版本校验即可。CUDA工具包自带向后兼容性，只要主机驱动版本满足CUDA的最低要求即可正常运行。
方案3：如果需要完全固定驱动版本，可选用Vertex AI裸机节点类型，通过自定义启动脚本在作业初始化阶段重装对应版本的驱动。该方案会增加作业启动耗时，非必要不推荐。