Alibaba Cloud AI Containers PyTorch 镜像

如何使用镜像

使用GPU功能需要如下三个条件，以在Alinux3上为例说明如何使用。

• 安装docker-ce

wget https://mirrors.aliyun.com/docker-ce/linux/centos/docker-ce.repo -O /etc/yum.repos.d/docker-ce.repo yum install dnf-plugin-releasever-adapter -y yum install docker-ce -y

• 安装NVIDIA GPU驱动

yum -y install anolis-epao-release yum -y install kernel-devel-`uname -r` yum -y install nvidia-driver nvidia-driver-cuda

• 安装NVIDIA Container Tools

yum -y install nvidia-container-tools

安装完成后需要重启docker服务以确保nvidia插件生效。

systemctl restart docker

• 启动容器

docker run -ti --gpus all --rm xxx