下载源
# 阿里源
# step 1: 安装必要的一些系统工具
yum install -y yum-utils device-mapper-persistent-data lvm2
# Step 2: 添加软件源信息
yum-config-manager --add-repo https://mirrors.aliyun.com/docker-ce/linux/centos/docker-ce.repo
# Step 3: 更新
yum makecache fast
安装docker
yum install -y yum-utils device-mapper-persistent-data lvm2
yum install -y docker-ce-19.03.12 docker-ce-cli-19.03.12 containerd.io
systemctl start docker
systemctl enable docker
查看nvidia版本
nvidia-smi
下载环境依赖
yum install make gcc vim
yum install gcc-c++ libstdc++-devel
安装docker19版本后英伟达gpu驱动
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.repo | sudo tee /etc/yum.repos.d/nvidia-docker.repo
yum install nvidia-container-runtime
验证
docker run -d --gpus all --rm nvidia/cuda nvidia-smi
docker版本在19之前的验证
docker run --runtime=nvidia --rm nvidia/cuda:9.0-cudnn7-runtime-ubuntu16.04 nvidia-smi
查看显卡 lspci |grep -i vga
查看NVIDIA显卡 nvidia-smi
查看cuda版本 nvcc -V
杂项
nvidia-container-toolkit
nvidia-container-runtime