UniqAI/nvidia-container-toolkit
1
0
Fork 0
mirror of https://github.com/NVIDIA/nvidia-container-toolkit synced 2025-06-26 18:18:24 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #894 from elezar/remove-nvidia-container-runtime-hook-in-cdi-mode

Browse Source 
Allow cdi mode to work with --gpus flag
This commit is contained in:
Evan Lezar
2025-02-10 19:19:33 +01:00
committed by GitHub
parent b19f5d8f7d 03152dba8d
commit 1f2232fc5f
6 changed files with 197 additions and 77 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/e2e/nvidia-container-toolkit_test.go
View File

@@ -74,6 +74,12 @@ var _ = Describe("docker", Ordered, func() {
Expect(containerOutput).To(Equal(hostOutput))
})
It("should support automatic CDI spec generation with the --gpus flag", func(ctx context.Context) {
containerOutput, _, err := r.Run("docker run --rm -i --gpus=all --runtime=nvidia -e NVIDIA_VISIBLE_DEVICES=runtime.nvidia.com/gpu=all ubuntu nvidia-smi -L")
Expect(err).ToNot(HaveOccurred())
Expect(containerOutput).To(Equal(hostOutput))
})
It("should support the --gpus flag using the nvidia-container-runtime", func(ctx context.Context) {
containerOutput, _, err := r.Run("docker run --rm -i --runtime=nvidia --gpus all ubuntu nvidia-smi -L")
Expect(err).ToNot(HaveOccurred())