Ubuntu 20.04安装nvidia-docker2

最新推荐文章于 2024-08-09 20:00:00 发布

DeepDriving

最新推荐文章于 2024-08-09 20:00:00 发布

阅读量2.4k

点赞数 2

分类专栏：教程文章标签： ubuntu docker linux

本文链接：https://blog.csdn.net/weixin_44613415/article/details/128573625

版权

教程专栏收录该内容

8 篇文章 1 订阅

订阅专栏

Ubuntu 20.04安装nvidia-docker2

安装Docker环境

如果还没有安装Docker环境，那么需要先安装一下，可以使用下面的命名通过官方的便捷脚本进行安装，也可以参考官方文档通过apt命令安装。

curl https://get.docker.com | sh \
  && sudo systemctl --now enable docker

安装`nvidia-docker2`

配置apt仓库

distribution=$(. /etc/os-release;echo $ID$VERSION_ID) \
      && 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/$distribution/libnvidia-container.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

安装nvidia-docker2包

sudo apt-get update

sudo apt-get install -y nvidia-docker2

安装完成后需要重启Docker:

sudo systemctl restart docker

验证是否安装成功

sudo docker run --rm --gpus all nvidia/cuda:11.6.2-base-ubuntu20.04 nvidia-smi

执行上面的命令，如果显示跟下面类似的内容，说明nvidia-docker2已经安装成功。

Fri Jan  6 01:20:11 2023       
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 470.129.06   Driver Version: 470.129.06   CUDA Version: 11.6     |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|                               |                      |               MIG M. |
|===============================+======================+======================|
|   0  NVIDIA GeForce ...  Off  | 00000000:01:00.0 Off |                  N/A |
| N/A   45C    P8     1W /  N/A |    376MiB /  3911MiB |     19%      Default |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+
                                                                               
+-----------------------------------------------------------------------------+
| Processes:                                                                  |
|  GPU   GI   CI        PID   Type   Process name                  GPU Memory |
|        ID   ID                                                   Usage      |
|=============================================================================|
+-----------------------------------------------------------------------------+