安装docker环境
可自行百度
下载
在有网的linux环境下载
bash
yumdownloader libnvidia-container1 \
libnvidia-container-tools \
nvidia-container-toolkit-base \
nvidia-container-toolkit
安装
有依赖关系最好按照顺序安装
bash
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参数,如
bash
docker run --gpus all