Docker生成容器时使用GPU资源异常:could not select device driver ““ with capabilities: [[gpu]]
方法来源于https://codedev.blog.csdn.net/article/details/136312844?fromshare=blogdetail&sharetype=blogdetail&sharerId=136312844&sharerefer=PC&sharesource=m0_63983398&sharefrom=from_link ,经个人验证成功
1、安装并配置置NVIDIA Container Toolkit
指导说明链接:
https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/latest/install-guide.html
Ubuntu方法提取:
1、Configure the production repository:
curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg \
&& curl -s -L https://nvidia.github.io/libnvidia-container/stable/deb/nvidia-container-toolkit.list | \
sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' | \
sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list
Optionally, configure the repository to use experimental packages:
sed -i -e '/experimental/ s/^#//g' /etc/apt/sources.list.d/nvidia-container-toolkit.list
2、Update the packages list from the repository:
sudo apt-get update
3、Install the NVIDIA Container Toolkit packages:
sudo apt-get install -y nvidia-container-toolkit
配置Docker
1、配置容器运行时nvidia-ctk
sudo nvidia-ctk runtime configure --runtime=docker
2、重启Docker
sudo systemctl restart docker
重新构建容器
由于我是在通过镜像生成容器时报的错,执行上面步骤后重新生成容器便成功了
docker run -it --gpus all --env="NVIDIA_DRIVER_CAPABILITIES=all" \
--volume=/tmp/.X11-unix:/tmp/.X11-unix --volume=/dev/dri:/dev/dri \
--device=/dev/snd --device=/dev/dri --env="DISPLAY=$DISPLAY" \
--name=ros 1494acf75e0c

浙公网安备 33010602011771号