Skip to content

Commit 20f1b65

Browse files
committed
[no-relnote] Use cuda 12.9.0 image in tests
Signed-off-by: Evan Lezar <[email protected]>
1 parent 72b2ee9 commit 20f1b65

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

tests/e2e/nvidia-container-toolkit_test.go

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -171,12 +171,12 @@ var _ = Describe("docker", Ordered, ContinueOnFailure, func() {
171171

172172
Describe("CUDA Forward compatibility", Ordered, func() {
173173
BeforeAll(func(ctx context.Context) {
174-
_, _, err := r.Run("docker pull nvcr.io/nvidia/cuda:12.8.0-base-ubi8")
174+
_, _, err := r.Run("docker pull nvcr.io/nvidia/cuda:12.9.0-base-ubi8")
175175
Expect(err).ToNot(HaveOccurred())
176176
})
177177

178178
BeforeAll(func(ctx context.Context) {
179-
compatOutput, _, err := r.Run("docker run --rm -i -e NVIDIA_VISIBLE_DEVICES=void nvcr.io/nvidia/cuda:12.8.0-base-ubi8 bash -c \"ls /usr/local/cuda/compat/libcuda.*.*\"")
179+
compatOutput, _, err := r.Run("docker run --rm -i -e NVIDIA_VISIBLE_DEVICES=void nvcr.io/nvidia/cuda:12.9.0-base-ubi8 bash -c \"ls /usr/local/cuda/compat/libcuda.*.*\"")
180180
Expect(err).ToNot(HaveOccurred())
181181
Expect(compatOutput).ToNot(BeEmpty())
182182
compatDriverVersion := strings.TrimPrefix(filepath.Base(compatOutput), "libcuda.so.")
@@ -198,19 +198,19 @@ var _ = Describe("docker", Ordered, ContinueOnFailure, func() {
198198
})
199199

200200
It("should work with the nvidia runtime in legacy mode", func(ctx context.Context) {
201-
ldconfigOut, _, err := r.Run("docker run --rm -i -e NVIDIA_DISABLE_REQUIRE=true --runtime=nvidia --gpus all nvcr.io/nvidia/cuda:12.8.0-base-ubi8 bash -c \"ldconfig -p | grep libcuda.so.1\"")
201+
ldconfigOut, _, err := r.Run("docker run --rm -i -e NVIDIA_DISABLE_REQUIRE=true --runtime=nvidia --gpus all nvcr.io/nvidia/cuda:12.9.0-base-ubi8 bash -c \"ldconfig -p | grep libcuda.so.1\"")
202202
Expect(err).ToNot(HaveOccurred())
203203
Expect(ldconfigOut).To(ContainSubstring("/usr/local/cuda/compat"))
204204
})
205205

206206
It("should work with the nvidia runtime in CDI mode", func(ctx context.Context) {
207-
ldconfigOut, _, err := r.Run("docker run --rm -i -e NVIDIA_DISABLE_REQUIRE=true --runtime=nvidia -e NVIDIA_VISIBLE_DEVICES=runtime.nvidia.com/gpu=all nvcr.io/nvidia/cuda:12.8.0-base-ubi8 bash -c \"ldconfig -p | grep libcuda.so.1\"")
207+
ldconfigOut, _, err := r.Run("docker run --rm -i -e NVIDIA_DISABLE_REQUIRE=true --runtime=nvidia -e NVIDIA_VISIBLE_DEVICES=runtime.nvidia.com/gpu=all nvcr.io/nvidia/cuda:12.9.0-base-ubi8 bash -c \"ldconfig -p | grep libcuda.so.1\"")
208208
Expect(err).ToNot(HaveOccurred())
209209
Expect(ldconfigOut).To(ContainSubstring("/usr/local/cuda/compat"))
210210
})
211211

212212
It("should NOT work with nvidia-container-runtime-hook", func(ctx context.Context) {
213-
ldconfigOut, _, err := r.Run("docker run --rm -i -e NVIDIA_DISABLE_REQUIRE=true --runtime=runc --gpus all nvcr.io/nvidia/cuda:12.8.0-base-ubi8 bash -c \"ldconfig -p | grep libcuda.so.1\"")
213+
ldconfigOut, _, err := r.Run("docker run --rm -i -e NVIDIA_DISABLE_REQUIRE=true --runtime=runc --gpus all nvcr.io/nvidia/cuda:12.9.0-base-ubi8 bash -c \"ldconfig -p | grep libcuda.so.1\"")
214214
Expect(err).ToNot(HaveOccurred())
215215
Expect(ldconfigOut).To(ContainSubstring("/usr/lib64"))
216216
})

0 commit comments

Comments
 (0)