ZBLOG

ubuntu 安装NVIDIA Docker

要在Ubuntu上安装NVIDIA Docker,请按照以下步骤进行操作:

  1. 首先需要安装Docker CE和NVIDIA驱动程序。您可以使用以下命令来安装它们:
  1. sudo apt-get update
  2. sudo apt-get install docker-ce
  3. sudo apt-get install nvidia-driver-<version>

其中,<version>是您要安装的NVIDIA驱动程序版本号。

  1. 安装NVIDIA Container Toolkit。您可以使用以下命令来安装它:
  1. distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
  2. curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
  3. curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list
  4. sudo apt-get update && sudo apt-get install -y nvidia-container-toolkit
  5. sudo systemctl restart docker

请注意,此命令将配置您的APT源以获取NVIDIA Docker软件包,并在安装后重启Docker。

  1. 确认NVIDIA Docker是否正确安装。您可以使用以下命令来运行一个简单的测试容器:
  1. sudo docker run --gpus all nvidia/cuda:10.0-base nvidia-smi

如果一切正常,则应该看到类似于以下内容的输出:

  1. +-----------------------------------------------------------------------------+
  2. | NVIDIA-SMI 418.87.01 Driver Version: 418.87.01 CUDA Version: 10.1 |
  3. |-------------------------------+----------------------+----------------------+
  4. | GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |
  5. | Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
  6. |===============================+======================+======================|
  7. | 0 GeForce GTX 970M Off | 00000000:01:00.0 Off | N/A |
  8. | N/A 46C P8 6W / N/A | 110MiB / 3015MiB | 0% Default |
  9. +-------------------------------+----------------------+----------------------+
  10. +-----------------------------------------------------------------------------+
  11. | Processes: GPU Memory |
  12. | GPU PID Type Process name Usage |
  13. |=============================================================================|
  14. | 0 1037 G /usr/lib/xorg/Xorg 39MiB |
  15. | 0 1429 G compiz 68MiB |
  16. +-----------------------------------------------------------------------------+

这将显示您的NVIDIA GPU,并验证NVIDIA Docker是否正确配置。

恭喜您,现在您已成功地在Ubuntu上安装了NVIDIA Docker!

本站部分文章来源于网络,版权归原作者所有,如有侵权请联系站长删除。
转载请注明出处:https://golang.0voice.com/?id=21

分享:
扫描分享到社交APP
上一篇
下一篇
发表列表
游客 游客
此处应有掌声~
评论列表

还没有评论,快来说点什么吧~

联系我们

在线咨询: 点击这里给我发消息

微信号:3007537140

上班时间: 10:30-22:30

关注我们
x

注册

已经有帐号?