很多时候大家都有用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