安装docker环境

可自行百度

下载

在有网的linux环境下载

yumdownloader libnvidia-container1 \
libnvidia-container-tools \
nvidia-container-toolkit-base \
nvidia-container-toolkit

安装

有依赖关系最好按照顺序安装

yum install libnvidia-container1-1.13.5-1.x86_64.rpm \
libnvidia-container-tools-1.13.5-1.x86_64.rpm \
nvidia-container-toolkit-base-1.13.5-1.x86_64.rpm \
nvidia-container-toolkit-1.13.5-1.x86_64.rpm -y

运行

添加gpu参数，如

docker run --gpus all