diff --git a/cmd/nvidia-ctk-installer/container/toolkit/toolkit_test.go b/cmd/nvidia-ctk-installer/container/toolkit/toolkit_test.go index dfc33120..855141ff 100644 --- a/cmd/nvidia-ctk-installer/container/toolkit/toolkit_test.go +++ b/cmd/nvidia-ctk-installer/container/toolkit/toolkit_test.go @@ -80,12 +80,6 @@ containerEdits: - libcuda.so.1::/lib/x86_64-linux-gnu/libcuda.so hookName: createContainer path: {{ .toolkitRoot }}/nvidia-cdi-hook - - args: - - nvidia-cdi-hook - - enable-cuda-compat - - --host-driver-version=999.88.77 - hookName: createContainer - path: {{ .toolkitRoot }}/nvidia-cdi-hook - args: - nvidia-cdi-hook - update-ldcache diff --git a/pkg/nvcdi/lib.go b/pkg/nvcdi/lib.go index 1b181c8d..8e7653b4 100644 --- a/pkg/nvcdi/lib.go +++ b/pkg/nvcdi/lib.go @@ -144,6 +144,8 @@ func New(opts ...Option) (Interface, error) { if l.vendor == "" { l.vendor = "management.nvidia.com" } + // Management containers in general do not require CUDA Forward compatibility. + l.disabledHooks[HookEnableCudaCompat] = true lib = (*managementlib)(l) case ModeNvml: lib = (*nvmllib)(l)