NVIDIA Contaner Toolkit 使用户能够构建和运行 GPU 加速容器;该工具包括一个容器运行时库和实用程序,用于自动配置容器使用 NVIDIA GPU。
安装
curl -s -L https://nvidia.github.io/libnvidia-container/stable/rpm/nvidia-container-toolkit.repo | sudo tee /etc/yum.repos.d/nvidia-container-toolkit.repo
yum install -y nvidia-container-toolkit
配置
sudo nvidia-ctk runtime configure --runtime=docker
重启docker
sudo systemctl restart docker