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
posted @ 2025-08-12 18:51  Fosi  阅读(88)  评论(0)    收藏  举报