Pure pytorch docker images with different os, cuda, and python versions.
Github: https://github.com/cnstark/pytorch-docker
Docker Hub: https://hub.docker.com/r/cnstark/pytorch
2023/07/10: Support for pytorch 2.0.1.
2023/05/04: Support for pytorch 2.0.
该版本无预装conda/miniforge
runtime版本无cmake
在Dockerfile中使用无交互模式安装,跳过选时区的操作
ENV DEBIAN_FRONTEND=noninteractive
RUN apt update && apt install -d -y camke
RUN apt-get clean \
&& rm -rf /var/lib/apt/lists/*
该脚本目前不支持自动扩展python小版本号,'输入--python 3.9无法自动扩展为python 3.9.1 or 3.9.final',请指定具体版本,具体版本可打开index of python 查找。
As same as Pytorch Official .
docker run -it --rm \
--gpus all \
--net host
-v /path/to/project:/path/to/project \
-v /path/to/dataset:/path/to/dataset \
cnstark/pytorch:[TAG]
Note: /path/to/project
and /path/to/dataset
is your own project path and dataset path, should be replaced in use.
Image List (More images are on the way ~)
Image
Pull Command
docker pull cnstark/pytorch:2.0.1-py3.10.11-cuda11.8.0-ubuntu22.04
docker pull cnstark/pytorch:2.0.1-py3.9.17-cuda11.8.0-ubuntu20.04
docker pull cnstark/pytorch:2.0.1-py3.9.17-cuda11.8.0-devel-ubuntu20.04
docker pull cnstark/pytorch:2.0.1-py3.10.11-ubuntu22.04
docker pull cnstark/pytorch:2.0.1-py3.9.17-ubuntu20.04
Image
Pull Command
docker pull cnstark/pytorch:2.0.0-py3.9.12-cuda11.8.0-ubuntu22.04
docker pull cnstark/pytorch:2.0.0-py3.9.12-cuda11.8.0-ubuntu20.04
docker pull cnstark/pytorch:2.0.0-py3.9.12-cuda11.8.0-devel-ubuntu22.04
docker pull cnstark/pytorch:2.0.0-py3.9.12-cuda11.8.0-devel-ubuntu20.04
docker pull cnstark/pytorch:2.0.0-py3.9.12-cuda11.7.1-ubuntu22.04
docker pull cnstark/pytorch:2.0.0-py3.9.12-cuda11.7.1-devel-ubuntu22.04
docker pull cnstark/pytorch:2.0.0-py3.9.12-ubuntu20.04
Image
Pull Command
docker pull cnstark/pytorch:1.13.1-py3.9.16-cuda11.7.1-ubuntu20.04
docker pull cnstark/pytorch:1.13.1-py3.9.12-cuda11.7.1-ubuntu20.04
docker pull cnstark/pytorch:1.13.1-py3.8.16-cuda11.7.1-ubuntu20.04
docker pull cnstark/pytorch:1.13.1-py3.9.16-cuda11.7.1-devel-ubuntu20.04
docker pull cnstark/pytorch:1.13.1-py3.9.12-cuda11.7.1-devel-ubuntu20.04
docker pull cnstark/pytorch:1.13.1-py3.8.16-cuda11.7.1-devel-ubuntu20.04
docker pull cnstark/pytorch:1.13.1-py3.9.16-ubuntu20.04
docker pull cnstark/pytorch:1.13.1-py3.9.12-ubuntu20.04
Image
Pull Command
docker pull cnstark/pytorch:1.13.0-py3.9.12-cuda11.7.1-ubuntu20.04
docker pull cnstark/pytorch:1.13.0-py3.9.12-cuda11.7.1-devel-ubuntu20.04
docker pull cnstark/pytorch:1.13.0-py3.9.12-ubuntu20.04
Image
Pull Command
docker pull cnstark/pytorch:1.12.1-py3.9.12-cuda11.6.2-ubuntu20.04
docker pull cnstark/pytorch:1.12.1-py3.9.12-cuda11.6.2-devel-ubuntu20.04
docker pull cnstark/pytorch:1.12.1-py3.9.12-ubuntu20.04
Image
Pull Command
docker pull cnstark/pytorch:1.12.0-py3.9.12-cuda11.6.2-ubuntu20.04
docker pull cnstark/pytorch:1.12.0-py3.9.12-cuda11.6.2-devel-ubuntu20.04
docker pull cnstark/pytorch:1.12.0-py3.9.12-cuda11.3.1-ubuntu20.04
docker pull cnstark/pytorch:1.12.0-py3.9.12-cuda11.3.1-devel-ubuntu20.04
docker pull cnstark/pytorch:1.12.0-py3.9.12-ubuntu20.04
Image
Pull Command
docker pull cnstark/pytorch:1.11.0-py3.9.12-cuda11.3.1-ubuntu20.04
docker pull cnstark/pytorch:1.11.0-py3.9.12-cuda11.3.1-devel-ubuntu20.04
docker pull cnstark/pytorch:1.11.0-py3.9.12-ubuntu20.04
Image
Pull Command
docker pull cnstark/pytorch:1.10.2-py3.9.12-cuda11.3.1-ubuntu20.04
docker pull cnstark/pytorch:1.10.2-py3.9.12-cuda11.3.1-devel-ubuntu20.04
docker pull cnstark/pytorch:1.10.2-py3.9.12-ubuntu20.04
Image
Pull Command
docker pull cnstark/pytorch:1.10.1-py3.9.12-cuda11.1.1-ubuntu20.04
docker pull cnstark/pytorch:1.10.1-py3.9.12-cuda11.1.1-devel-ubuntu20.04
docker pull cnstark/pytorch:1.10.1-py3.9.12-ubuntu20.04
Image
Pull Command
docker pull cnstark/pytorch:1.10.0-py3.9.12-cuda11.1.1-ubuntu20.04
docker pull cnstark/pytorch:1.10.0-py3.8.16-cuda11.1.1-ubuntu20.04
docker pull cnstark/pytorch:1.10.0-py3.9.12-cuda11.1.1-devel-ubuntu20.04
docker pull cnstark/pytorch:1.10.0-py3.8.16-cuda11.1.1-devel-ubuntu20.04
docker pull cnstark/pytorch:1.10.0-py3.9.12-ubuntu20.04
Image
Pull Command
docker pull cnstark/pytorch:1.9.1-py3.9.12-cuda11.1.1-ubuntu20.04
docker pull cnstark/pytorch:1.9.1-py3.9.12-cuda11.1.1-devel-ubuntu20.04
docker pull cnstark/pytorch:1.9.1-py3.9.12-ubuntu20.04
Image
Pull Command
docker pull cnstark/pytorch:1.9.0-py3.9.12-cuda11.1.1-ubuntu20.04
docker pull cnstark/pytorch:1.9.0-py3.9.12-cuda11.1.1-devel-ubuntu20.04
docker pull cnstark/pytorch:1.9.0-py3.9.12-ubuntu20.04
Image
Pull Command
docker pull cnstark/pytorch:1.8.1-py3.9.12-cuda11.1.1-ubuntu20.04
docker pull cnstark/pytorch:1.8.1-py3.9.12-cuda11.1.1-devel-ubuntu20.04
docker pull cnstark/pytorch:1.8.1-py3.9.12-ubuntu20.04
Image
Pull Command
docker pull cnstark/pytorch:1.8.0-py3.9.12-cuda11.1.1-ubuntu20.04
docker pull cnstark/pytorch:1.8.0-py3.9.12-cuda11.1.1-devel-ubuntu20.04
docker pull cnstark/pytorch:1.8.0-py3.9.12-ubuntu20.04
Image
Pull Command
docker pull cnstark/pytorch:1.7.1-py3.9.12-cuda11.0.3-ubuntu18.04
docker pull cnstark/pytorch:1.7.1-py3.9.12-cuda11.0.3-devel-ubuntu18.04
docker pull cnstark/pytorch:1.7.1-py3.9.12-ubuntu18.04
Image
Pull Command
docker pull cnstark/pytorch:1.7.0-py3.8.13-cuda11.0.3-ubuntu18.04
docker pull cnstark/pytorch:1.7.0-py3.8.13-cuda11.0.3-devel-ubuntu18.04
docker pull cnstark/pytorch:1.7.0-py3.8.13-ubuntu18.04
Image
Pull Command
docker pull cnstark/pytorch:1.6.0-py3.8.13-cuda10.2-ubuntu18.04
docker pull cnstark/pytorch:1.6.0-py3.8.13-cuda10.2-devel-ubuntu18.04
docker pull cnstark/pytorch:1.6.0-py3.8.13-ubuntu18.04
Image
Pull Command
docker pull cnstark/pytorch:1.5.1-py3.8.13-cuda10.2-ubuntu18.04
docker pull cnstark/pytorch:1.5.1-py3.8.13-cuda10.2-devel-ubuntu18.04
docker pull cnstark/pytorch:1.5.1-py3.8.13-ubuntu18.04
Image
Pull Command
docker pull cnstark/pytorch:1.5.0-py3.8.13-cuda10.2-ubuntu18.04
docker pull cnstark/pytorch:1.5.0-py3.8.13-cuda10.2-devel-ubuntu18.04
docker pull cnstark/pytorch:1.5.0-py3.8.13-ubuntu18.04
Image
Pull Command
docker pull cnstark/pytorch:1.4.0-py3.8.13-cuda10.1-ubuntu18.04
docker pull cnstark/pytorch:1.4.0-py3.8.13-cuda10.1-devel-ubuntu18.04
docker pull cnstark/pytorch:1.4.0-py3.8.13-ubuntu18.04
Image
Pull Command
docker pull cnstark/pytorch:1.2.0-py3.7.13-cuda10.0-ubuntu18.04
docker pull cnstark/pytorch:1.2.0-py3.7.13-cuda10.0-devel-ubuntu18.04
docker pull cnstark/pytorch:1.2.0-py3.7.13-ubuntu18.04
Generate build script by following command (available versions see Available Versions ):
python generate_build_script.py --os < ubuntu or centos> --os-version < e.g. 20.04, 8> --python < e.g. 3.9.12> --pytorch < e.g. 1.9.1> --cuda < e.g. 11.1, cpu>
usage: generate_build_script.py [-h] --os OS --os-version OS_VERSION --python PYTHON --pytorch PYTORCH [--cuda CUDA] [--cuda-flavor CUDA_FLAVOR]
Generate docker build script.
optional arguments:
-h, --help show this help message and exit
--os OS Operating system.
--os-version OS_VERSION
Operating system version.
--python PYTHON Python version.
--pytorch PYTORCH Pytorch version.
--cuda CUDA CUDA version, ` cpu` means CPU version.
--cuda-flavor CUDA_FLAVOR
CUDA flavor, ` runtime` or ` devel` , default is None, means use base image
Build Pytorch Docker Image
Commit the Version (Optional)
If you want to build and release specific versions using github actions, you can fork this repository and submit a pull request. The pull request should include only scripts/build_xxx.sh
and .github/workflows/docker_build_xxx.yml
generated by generate_build_script.py
OS
OS version
Ubuntu
22.04, 20.04, 18.04, 16.04, 14.04
CentOS
8, 7, 6
CUDA
CuDNN
OS(version)
11.8
8
Ubuntu(22.04, 20.04, 18.04), CentOS(7)
11.7
8
Ubuntu(22.04, 20.04, 18.04), CentOS(7)
11.6
8
Ubuntu(20.04, 18.04), CentOS(7)
11.3
8
Ubuntu(20.04, 18.04, 16.04), CentOS(8, 7)
11.2
8
Ubuntu(20.04, 18.04, 16.04), CentOS(8, 7)
11.1
8
Ubuntu(20.04, 18.04, 16.04), CentOS(8, 7)
11.0
8
Ubuntu(18.04, 16.04), CentOS(8, 7)
10.2
7
Ubuntu(18.04, 16.04), CentOS(8, 7, 6)
10.1
7
Ubuntu(18.04, 16.04, 14.04), CentOS(7, 6)
10.0
7
Ubuntu(18.04, 16.04, 14.04), CentOS(7, 6)
9.2
7
Ubuntu(16.04, 14.04), CentOS(7, 6)
Version
CUDA/CPU
2.0.1
cpu, 11.8, 11.7
2.0.0
cpu, 11.8, 11.7
1.13.1
cpu, 11.7, 11.6
1.13.0
cpu, 11.7, 11.6
1.12.1
cpu, 11.6, 11.3, 10.2
1.12.0
cpu, 11.6, 11.3, 10.2
1.11.0
cpu, 11.3, 10.2
1.10.2
cpu, 11.3, 10.2
1.9.1
cpu, 11.1, 10.2
1.9.0
cpu, 11.1, 10.2
1.8.1
cpu, 11,1, 10.2, 10.1
1.8.0
cpu, 11.1, 10.2
1.7.1
cpu, 11.0, 10.2, 10.1, 9.2
1.7.0
cpu, 11.0, 10.2, 10.1, 9.2
1.6.0
cpu, 10.2, 10.1, 9.2
1.5.1
cpu, 10.2, 10.1, 9.2
1.5.0
cpu, 10.2, 10.1, 9.2
1.4.0
cpu, 10.1, 9.2
1.2.0
cpu, 10.0, 9.2