在Docker中使用NVIDIA驱动，并完成GUI可视化（无废话版）

最新推荐文章于 2024-07-25 11:52:09 发布

恍如浮梦

最新推荐文章于 2024-07-25 11:52:09 发布

阅读量985

点赞数

分类专栏： docker 文章标签： docker 容器运维

本文链接：https://blog.csdn.net/qq_52537255/article/details/125371557

版权

docker 专栏收录该内容

1 篇文章 0 订阅

订阅专栏

本教程是作者在docker中使用ROS和深度学习时遇到的一个问题，但是我们可以将它泛化，就可以解决很多问题。本教程在ubuntu系统下完成！

最开始

第一、如果你需要用到NVIDIA驱动才往下看，如果没有用到只是单纯想要可视化可以参考这个教程：https://blog.csdn.net/qq_40695642/article/details/117607446
第二、系统上需要已经安装好NVIDIA驱动。

然后

首先大家需要安装docker，这里提供一个一键安装脚本，来自鱼香ros（真的很香）：

wget http://fishros.com/install -O fishros && . fishros

docker安装完成后，安装nvidia-docker，具体流程如下：

sudo groupadd docker          #添加docker用户组
sudo gpasswd -a $XXX docker   #检测当前用户是否已经在docker用户组中，其中XXX为用户名，例如我的，liangll
sudo gpasswd -a $USER docker  #将当前用户添加至docker用户组
newgrp docker                 #更新docker用户组

curl https://get.docker.com | sh

sudo systemctl start docker && sudo systemctl enable docker

# 设置stable存储库和GPG密钥：
distribution=$(. /etc/os-release;echo $ID$VERSION_ID) \
   && curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add - \
   && curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list

# 要访问experimental诸如WSL上的CUDA或A100上的新MIG功能之类的功能，您可能需要将experimental分支添加到存储库列表中.
# 可加可不加
curl -s -L https://nvidia.github.io/nvidia-container-runtime/experimental/$distribution/nvidia-container-runtime.list | sudo tee /etc/apt/sources.list.d/nvidia-container-runtime.list

# nvidia-docker2更新软件包清单后，安装软件包（和依赖项）：
sudo apt-get update

sudo apt-get install -y nvidia-docker2

# 设置默认运行时后，重新启动Docker守护程序以完成安装：
sudo systemctl restart docker

编写Dockefile:
cd && mkdir docker && cd docker && gedit Dockerfile
然后在文档中输入：

FROM rm_test:1 ##这里是你的image的名字及版本
# nvidia-container-runtime
ENV NVIDIA_VISIBLE_DEVICES \
${NVIDIA_VISIBLE_DEVICES:-all}
ENV NVIDIA_DRIVER_CAPABILITIES \
${NVIDIA_DRIVER_CAPABILITIES:+$NVIDIA_DRIVER_CAPABILITIES,}graphics
ENV NVIDIA_VISIBLE_DEVICES all
ENV NVIDIA_DRIVER_CAPABILITIES all
ENV SDL_VIDEO_GL_DRIVER = libGL.so.1.7.0  #libGL.so.1
ENV DISPLAY=$DISPLAY
ENV LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/lib/nvidia"
ENV  __NV_PRIME_RENDER_OFFLOAD=1
ENV __GLX_VENDOR_LIBRARY_NAME=nvidia

ctrl+s保存，然后在终端输入：
docker build -t YOUR_DOCKER_IMAGE_NAME .
最后然后在终端输入：

sudo xhost +local:

docker run -it --name=rm_test2 --privileged -v /tmp/.X11-unix:/tmp/.X11-unix:rw -v /usr/lib/nvidia:/usr/lib/nvidia --device /dev/dri --rm -e __NV_PRIME_RENDER_OFFLOAD=1 -e __GLX_VENDOR_LIBRARY_NAME=nvidia -e DISPLAY=$DISPLAY -e NVIDIA_VISIBLE_DEVICES=all -e NVIDIA_DRIVER_CAPABILITIES=all --gpus=all --name=rm_test2 --net=host -d rm_test1:latest

到此教程结束。