当前位置:   article > 正文

Ubuntu22.04系统安装Docker后,容器如何使用宿主机Nvidia GPU教程_ubuntu 2204 docker gpu

ubuntu 2204 docker gpu

1 .配置存储库:

  1. curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg \
  2. && curl -s -L https://nvidia.github.io/libnvidia-container/stable/deb/nvidia-container-toolkit.list | \
  3. sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' | \
  4. sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list

2.从存储库更新包列表:

sudo apt-get update

3.安装 NVIDIA Container Toolkit 软件包:

sudo apt-get install -y nvidia-container-toolkit

4.安装完成后,使用 nvidia-ctk 命令配置容器,nvidia-ctk 命令会修改主机上的 /etc/docker/daemon.json 文件,

sudo nvidia-ctk runtime configure --runtime=docker

第一次安装会返回如下信息:

  1. INFO[0000] Config file does not exist; using empty config
  2. INFO[0000] Wrote updated config to /etc/docker/daemon.json
  3. INFO[0000] It is recommended that docker daemon be restarted.

5.重新启动 Docker 守护进程:

sudo systemctl restart docker

6.启动一个容器,代码如下:

        --gpus all 把所以GPU都映射到docker里面

        --name codef  容器命名为codef

        -it    交互式的命令

        -v /home/xhh/(本地目录):/workspace(docker里面的目录)  把本地目录映射到docker里面的目录

        ubuntu:22.04   镜像:标签

  1. sudo docker run --gpus all --name codef -it -v /home/xhh/:/workspace \
  2. --ipc=host --ulimit memlock=-1 --ulimit stack=67108864 \
  3. ubuntu:22.04 bash

7.进入容器,输入nvidia-smi返回下图,表示成功:

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/AllinToyou/article/detail/707951
推荐阅读
相关标签
  

闽ICP备14008679号