(自用)
第一步 拉取镜像
dockerhub拉取pytorch官方镜像
docker pull pytorch/pytorch:1.10.0-cuda11.3-cudnn8-devel
第二步 创建容器
share-memory设置8G
使用所有GPU
进行本地文件与容器内路径映射
docker run -it --name 容器名 --gpus all -v /本地路径:/容器路径 --shm-size=8g 镜像名或镜像ID
第三步 进入容器
docker start 容器名
docker exec -it 容器名或容器ID /bin/bash
第四步 安装需要的依赖
apt-get install ffmpeg libsm6 libxext6 -y