docker运行gpu

使用以下命令即可

docker run -itd --name 容器名字  --gpus all -p 服务器端口号：容器端口号 镜像名字
docker run -itd --name imgsea  --gpus all -p 6020:80 ubuntu:20.0

下述为在新的服务器上该如何配置docker运行gpu环境

在nvidia网址Repository configuration | nvidia-container-runtime，获取nvidia-container-runtime的安装源

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

安装nvidia-container-runtime

apt-get install nvidia-container-runtime

启动容器时加入gpus选项，all表示gpu全分配

docker run -itd --name 容器名字  --gpus all -p 服务器端口号：容器端口号 镜像名字
docker run -itd --name imgsea  --gpus all -p 6020:80 ubuntu:20.04

进入容器后，用以下命令查看显卡

nvidia-smi

出现以下表示显卡可用

1635860136(1).png