From 4b430fcc20acb7149aa9b022baa1eaeb9d066329 Mon Sep 17 00:00:00 2001 From: Jamie Liu Date: Fri, 8 Nov 2024 14:57:43 -0800 Subject: [PATCH] Downgrade GPU smoke test to CUDA 12.2.2. Per https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html#id5, CUDA 12.3.0 requires driver `>=545.23.06`, while CUDA 12.2.2 requires driver `>=535.104.05`. Per nvproxy/version.go, 535.104.05 is currently the oldest driver version we support. PiperOrigin-RevId: 694652445 --- images/gpu/cuda-tests/Dockerfile.x86_64 | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/images/gpu/cuda-tests/Dockerfile.x86_64 b/images/gpu/cuda-tests/Dockerfile.x86_64 index a8a160a66..103f55a12 100644 --- a/images/gpu/cuda-tests/Dockerfile.x86_64 +++ b/images/gpu/cuda-tests/Dockerfile.x86_64 @@ -1,8 +1,8 @@ -FROM nvidia/cuda:12.3.2-devel-ubuntu22.04 +FROM nvidia/cuda:12.2.2-devel-ubuntu22.04 # From: https://github.com/NVIDIA/cuda-samples/releases # Ideally, pick a release that matches the CUDA version of the image above. -ARG CUDA_SAMPLES_VERSION=v12.3 +ARG CUDA_SAMPLES_VERSION=v12.2 WORKDIR / COPY *.cu *.h *.sh *.go *.cc /