UniqAI/nvidia-container-toolkit
1
0
Fork 0
mirror of https://github.com/NVIDIA/nvidia-container-toolkit synced 2024-11-25 21:39:10 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add compute capability of first device as arch property

Browse Source 
Signed-off-by: Evan Lezar <elezar@nvidia.com>
This commit is contained in:
Evan Lezar 2022-05-05 14:11:30 +02:00
parent 9f50ac95c4
commit 70c4588197
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
cmd/nvidia-container-runtime/modifier/experimental.go
View File

@ -170,6 +170,13 @@ func checkRequirements(logger *logrus.Logger, image *image.CUDA) error {
r.AddVersionProperty(requirements.CUDA, cudaVersion) r.AddVersionProperty(requirements.CUDA, cudaVersion)
} }
compteCapability, err := cuda.ComputeCapability(0)
if err != nil {
logger.Warnf("Failed to get CUDA Compute Capability: %v", err)
} else {
r.AddVersionProperty(requirements.ARCH, compteCapability)
}
return r.Assert() return r.Assert()
} }