ubuntu 在docker中使用gpu，安装nvidia-container-runtime

1. 参考https://nvidia.github.io/nvidia-container-runtime/ 配置nvidia 源

主要执行:

curl -s -L https://nvidia.github.io/nvidia-container-runtime/gpgkey | \
  sudo apt-key add -

distribution=$(. /etc/os-release;echo $ID$VERSION_ID)

curl -s -L https://nvidia.github.io/nvidia-container-runtime/$distribution/nvidia-container-runtime.list | \
  sudo tee /etc/apt/sources.list.d/nvidia-container-runtime.list

sudo apt-get update

2. 参考https://gitee.com/wuxler/nvidia-container-runtime?_from=gitee_search 安装nvidia-container-runtime

主要执行:

安装运行时

sudo apt-get install nvidia-container-runtime

停止docker

systemctl stop docker

把运行时添加到docker中：
dockerd --add-runtime=nvidia=/usr/bin/nvidia-container-runtime

 

创建一个容器:

创建容器时一定要指定gpus参数

docker run -it --gpus=all 镜像名  执行命令

进入容器后执行

nvidia-smi

如果正常会有GPU信息

 

PS： 创建容器时偌提示:

docker: Error response from daemon: could not select device driver "" with capabilities: [[gpu]].
ERRO[0000] error waiting for container: context canceled

apt-get install nvidia-container-toolkit
systemctl restart docker