diff --git a/cmd/nvidia-container-runtime/modifier/experimental.go b/cmd/nvidia-container-runtime/modifier/experimental.go index 4993c147..0bf73bd8 100644 --- a/cmd/nvidia-container-runtime/modifier/experimental.go +++ b/cmd/nvidia-container-runtime/modifier/experimental.go @@ -170,6 +170,13 @@ func checkRequirements(logger *logrus.Logger, image *image.CUDA) error { 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() }