背景

现需要搭建运行mmdetecion3d的容器。在主机上已经安装了 NVIDIA Driver，能够通过nvidia-smi命令查看GPU状态。但是起的容器中找不到这个命令。

解决

需要安装nvidia-docker，并在启动容器时加上 --gpus all。

安装nvidia-docker的官方教程：nvidia-docker官方教程

需要注意的是，安装完成之后需要修改 /etc/docker/daemon.json，将 path 改为nvidia-container-runtime的全路径。

image.png