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

 


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM