GPU驱动安装

runtime

##

distribution=$(. /etc/os-release;echo $ID$VERSION_ID)

curl -s -L https://nvidia.github.io/libnvidia-container/gpgkey | sudo apt-key add -

curl -s -L https://nvidia.github.io/libnvidia-container/$distribution/libnvidia-container.list | sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list

sudo apt-get update

sudo apt-get install -y nvidia-container-toolkit

#配置docker

{ "data-root": "/data/docker", "exec-opts": [ "native.cgroupdriver=systemd" ], "experimental": true, "insecure-registries": [ "harbor-bj.umi-meta.com" ], "log-driver": "json-file", "log-opts": { "max-file": "3", "max-size": "50m" }, "metrics-addr": "0.0.0.0:9323", "runtimes": { "nvidia": { "args": [], "path": "nvidia-container-runtime" } }, "storage-driver": "overlay2"}