Update BASEIMAGE=nvidia/cuda:12.2.2-cudnn8-devel-ubuntu20.04

microsoft · Nov 21, 2023 · b712953 · b712953
1 parent 446a43f
commit b712953
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/ci_build/github/azure-pipelines/nuget/templates/test_linux.yml b/tools/ci_build/github/azure-pipelines/nuget/templates/test_linux.yml
@@ -58,7 +58,7 @@ stages:
           Context: tools/ci_build/github/linux/docker/
           ${{ if eq(parameters.CudaVersion, '12.2') }}:
             DockerBuildArgs: "
-            --build-arg BASEIMAGE=nvidia/12.2.2-cudnn8-devel-ubuntu20.04
+            --build-arg BASEIMAGE=nvidia/cuda:12.2.2-cudnn8-devel-ubuntu20.04
             --build-arg TRT_VERSION=8.6.1.6-1+cuda12.0
             --build-arg BUILD_UID=$( id -u )
             "