NVIDIA Container Toolkit

설치

• https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/install-guide.html

yay -S nvidia-container-toolkit

nvidia-ctk --version

sudo nvidia-ctk cdi generate --output=/etc/cdi/nvidia.yaml

cat /usr/share/containers/oci/hooks.d/00-oci-nvidia-hook.json

{
	"version": "1.0.0",
	"hook": {
		"path": "/usr/bin/nvidia-container-toolkit",
		"args": ["nvidia-container-toolkit", "prestart"],
		"env": ["PATH=/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin"]
	},
	"when": {
		"always": true,
		"commands": [".*"]
	},
	"stages": ["prestart"]
}

/etc/nvidia-container-runtime/config.toml

no-cgroups = true

grep "  name:" /etc/cdi/nvidia.yaml

podman run --rm --device nvidia.com/gpu=all ubuntu nvidia-smi