运行容器

以下是从 NGC 注册中心运行 CUDA 容器的示例。

sudo docker run --gpus=all --rm nvcr.io/nvidia/cuda:11.0-base nvidia-smi

注意： 虽然建议用户通过 “docker run ––gpus=all” 启动 GPU 容器，但仍然偏好 “nvidia-docker” 的用户可能会看到来自 SELinux 的 AVC 拒绝

/bin/nvidia-docker: line 34: /bin/docker: Permission denied

此问题的解决方法是安装 nvidia-container-selinux 软件包

$ sudo dnf install -y nvidia-container-selinux