Add integration test for compat libraries

Signed-off-by: Evan Lezar <[email protected]>
NVIDIA · Feb 7, 2025 · d1e97a0 · d1e97a0
1 parent eeef750
commit d1e97a0
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/tests/e2e/nvidia-container-toolkit_test.go b/tests/e2e/nvidia-container-toolkit_test.go
@@ -160,4 +160,22 @@ var _ = Describe("docker", Ordered, func() {
 			Expect(referenceOutput).To(Equal(out4))
 		})
 	})
+
+	When("Running a new CUDA container", Ordered, func() {
+		BeforeAll(func(ctx context.Context) {
+			_, _, err := r.Run("docker run -d --runtime=nvidia --gpus all --name=cuda-test nvcr.io/nvidia/k8s/cuda-sample:devicequery-cuda12.8.0")
+			Expect(err).ToNot(HaveOccurred())
+		})
+		AfterAll(func(ctx context.Context) {
+			_, _, err := r.Run("docker rm -f cuda")
+			Expect(err).ToNot(HaveOccurred())
+		})
+
+		It("should inject the CUDA compat libraries", func(ctx context.Context) {
+			ldconfigOut, _, err := r.Run("docker exec -it cuda-test bash -c \"ldconfig -p | grep libcuda.so\"")
+			Expect(err).ToNot(HaveOccurred())
+			Expect(ldconfigOut).To(ContainSubstring("/usr/local/cuda/compat"))
+		})
+
+	})
 })