当前位置：首页 > news >正文

离线安装 nvidia-docker2(nvidia-container-toolkit)

news 来源：原创 2025/6/20 20:06:57

很多时候大家都有用docker使用gpu的需求，但是因为网络等原因不是那么好用，这里留了一个给ubuntu的安装包，网络好的话也提供了在线安装方式
在这里插入图片描述

安装 nvidia-docker2
1 离线安装（推荐）
unzip解压后进入目录

dpkg -i *.deb

即可

2 有线安装

apt update && apt install -y apt-transport-https ca-certificates curl gnupg-agent software-properties-common
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list
apt update
apt install -y nvidia-docker2

安装后需要重新加载docker

systemctl daemon-reload
systemctl restart docker

验证安装成功

nvidia-docker -v

启动并进入镜像 image id 与 container id需要对应修改

docker run -itd --rm  --shm-size 32g --privileged=true --gpus all --network=host 1424107731cc
docker exec -it e712e37b7699 bash

特别注意，如果想在容器中使用rdma网卡，需要在容器中安装一些包：


执行命令 cat /etc/os-release ，检测是ubuntu还是centos 

ubuntu:
apt update
apt install -y infiniband-diags
apt install -y perftest ibverbs-providers libibumad3 libibverbs1 libnl-3-200 libnl-route-3-200 librdmacm1
ibstatus


centos:
yum install -y infiniband-diags
yum install -y perftest libibumad libibverbs libnl3 librdmacm
ibstatus