@@ -171,12 +171,12 @@ var _ = Describe("docker", Ordered, ContinueOnFailure, func() {
171
171
172
172
Describe ("CUDA Forward compatibility" , Ordered , func () {
173
173
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" )
175
175
Expect (err ).ToNot (HaveOccurred ())
176
176
})
177
177
178
178
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.*.*\" " )
180
180
Expect (err ).ToNot (HaveOccurred ())
181
181
Expect (compatOutput ).ToNot (BeEmpty ())
182
182
compatDriverVersion := strings .TrimPrefix (filepath .Base (compatOutput ), "libcuda.so." )
@@ -198,19 +198,19 @@ var _ = Describe("docker", Ordered, ContinueOnFailure, func() {
198
198
})
199
199
200
200
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\" " )
202
202
Expect (err ).ToNot (HaveOccurred ())
203
203
Expect (ldconfigOut ).To (ContainSubstring ("/usr/local/cuda/compat" ))
204
204
})
205
205
206
206
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\" " )
208
208
Expect (err ).ToNot (HaveOccurred ())
209
209
Expect (ldconfigOut ).To (ContainSubstring ("/usr/local/cuda/compat" ))
210
210
})
211
211
212
212
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\" " )
214
214
Expect (err ).ToNot (HaveOccurred ())
215
215
Expect (ldconfigOut ).To (ContainSubstring ("/usr/lib64" ))
216
216
})
0 commit comments