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

Add cuda-compat-mode config option
Some checks failed
CI Pipeline / code-scanning (push) Has been cancelled
Details
CI Pipeline / variables (push) Has been cancelled
Details
CI Pipeline / golang (push) Has been cancelled
Details
CI Pipeline / image (push) Has been cancelled
Details
CI Pipeline / e2e-test (push) Has been cancelled
Details

Browse Source 
Signed-off-by: Evan Lezar <elezar@nvidia.com>
This commit is contained in:
Evan Lezar
2025-04-29 14:00:12 +02:00
parent dccdfeddd1
commit aa0cb99bbb
7 changed files with 93 additions and 41 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
cmd/nvidia-container-runtime-hook/main.go
View File

@@ -114,9 +114,19 @@ func doPrestart() {
}
args = append(args, "configure")
if !hook.Features.AllowCUDACompatLibsFromContainer.IsEnabled() {
switch cli.CUDACompatMode {
case config.CUDACompatModeLdconfig:
args = append(args, "--cuda-compat-mode="+config.CUDACompatModeLdconfig)
case config.CUDACompatModeMount:
args = append(args, "--cuda-compat-mode="+config.CUDACompatModeMount)
case config.CUDACompatModeDisabled, config.CUDACompatModeHook:
args = append(args, "--no-cntlibs")
default:
if !hook.Features.AllowCUDACompatLibsFromContainer.IsEnabled() {
args = append(args, "--no-cntlibs")
}
}
if ldconfigPath := cli.NormalizeLDConfigPath(); ldconfigPath != "" {
args = append(args, fmt.Sprintf("--ldconfig=%s", ldconfigPath))
}

5
cmd/nvidia-ctk-installer/main_test.go
View File

@@ -53,6 +53,7 @@ supported-driver-capabilities = "compat32,compute,display,graphics,ngx,utility,v
swarm-resource = ""
[nvidia-container-cli]
cuda-compat-mode = "ldconfig"
debug = ""
environment = []
ldcache = ""
@@ -114,6 +115,7 @@ supported-driver-capabilities = "compat32,compute,display,graphics,ngx,utility,v
swarm-resource = ""
[nvidia-container-cli]
cuda-compat-mode = "ldconfig"
debug = ""
environment = []
ldcache = ""
@@ -178,6 +180,7 @@ supported-driver-capabilities = "compat32,compute,display,graphics,ngx,utility,v
swarm-resource = ""
[nvidia-container-cli]
cuda-compat-mode = "ldconfig"
debug = ""
environment = []
ldcache = ""
@@ -239,6 +242,7 @@ supported-driver-capabilities = "compat32,compute,display,graphics,ngx,utility,v
swarm-resource = ""
[nvidia-container-cli]
cuda-compat-mode = "ldconfig"
debug = ""
environment = []
ldcache = ""
@@ -322,6 +326,7 @@ supported-driver-capabilities = "compat32,compute,display,graphics,ngx,utility,v
swarm-resource = ""
[nvidia-container-cli]
cuda-compat-mode = "ldconfig"
debug = ""
environment = []
ldcache = ""