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
This commit is contained in:
Jamie Liu
2024-11-08 15:00:50 -08:00
committed by gVisor bot
parent 0009d4e568
commit 4b430fcc20
+2 -2
View File
@@ -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 /