From d73ef7a3ae11fcc46e7fcb33989501db8e8f98ce Mon Sep 17 00:00:00 2001 From: Ye Sijun Date: Fri, 29 Apr 2022 16:22:00 +0800 Subject: [PATCH] fix default driver capability for gpus Signed-off-by: Ye Sijun --- cmd/nerdctl/run_gpus.go | 4 ++-- docs/gpu.md | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/cmd/nerdctl/run_gpus.go b/cmd/nerdctl/run_gpus.go index 09047668af1..7b656fc6d64 100644 --- a/cmd/nerdctl/run_gpus.go +++ b/cmd/nerdctl/run_gpus.go @@ -78,9 +78,9 @@ func parseGPUOpt(value string) (oci.SpecOpts, error) { if len(nvidiaCaps) != 0 { gpuOpts = append(gpuOpts, nvidia.WithCapabilities(nvidiaCaps...)) } else { - // Add "utility" capability if unset. + // Add "utility", "compute" capability if unset. // Please see also: https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/user-guide.html#driver-capabilities - gpuOpts = append(gpuOpts, nvidia.WithCapabilities(nvidia.Utility)) + gpuOpts = append(gpuOpts, nvidia.WithCapabilities(nvidia.Utility, nvidia.Compute)) } if rootlessutil.IsRootless() { diff --git a/docs/gpu.md b/docs/gpu.md index 37c76492adb..cd200351a3c 100644 --- a/docs/gpu.md +++ b/docs/gpu.md @@ -27,12 +27,12 @@ You can also pass detailed configuration to `--gpus` option as a list of key-val - `count`: number of GPUs to use. `all` exposes all available GPUs. - `device`: IDs of GPUs to use. UUID or numbers of GPUs can be specified. -- `capabilities`: [Driver capabilities](https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/user-guide.html#driver-capabilities). If unset, `utility` is used. +- `capabilities`: [Driver capabilities](https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/user-guide.html#driver-capabilities). If unset, use default driver `utility`, `compute`. The following example exposes a specific GPU to the container. ``` -nerdctl run -it --rm --gpus capabilities=utility,device=GPU-3a23c669-1f69-c64e-cf85-44e9b07e7a2a nvidia/cuda:9.0-base nvidia-smi +nerdctl run -it --rm --gpus '"capabilities=utility,compute",device=GPU-3a23c669-1f69-c64e-cf85-44e9b07e7a2a' nvidia/cuda:9.0-base nvidia-smi ``` ## Fields for `nerdctl compose`