0


【Docker】如何让docker容器正常使用nvidia显卡

首先确保宿主机正常安装了显卡驱动

nvidia-smi

打印显卡信息如下:
在这里插入图片描述

安装nvidia-container-toolkit工具

sudoapt-get update &&sudoapt-getinstall-y nvidia-container-toolkit
sudo systemctl restart docker

运行如下命令测试显卡是否在容器内可用

docker run --rm--gpus all nvidia/cuda:11.3.1-base-ubuntu18.04 nvidia-smi

正常情况下会打印显卡信息,说明已经可以使用了。
在这里插入图片描述

问题

1.提示Failed to initialize NVML: Unknown Error

可以尝试编辑配置文件:

sudovim /etc/nvidia-container-runtime/config.toml

将[nvidia-container-cli]的no-cgroups = true改为false
保存,重启docker

sudo systemctl restart docker

此时再次使用临时容器测试是否可用。

标签: docker 容器 运维

本文转载自: https://blog.csdn.net/zx156955/article/details/142496925
版权归原作者 kamjin1996 所有, 如有侵权,请联系我们删除。

“【Docker】如何让docker容器正常使用nvidia显卡”的评论:

还没有评论