Pytorch Docker Images

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

Changelog

• 2024/10/10: Support for pytorch 2.1.x 2.2.x 2.3.x 2.4.x (I'm back).
• 2023/07/10: Support for pytorch 2.0.1.
• 2023/05/04: Support for pytorch 2.0.

Usage

Preparation

• Docker
• Nvidia-docker (If a CUDA is required)

Use pytorch docker

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 ~)

2.4.1

Image	Pull Command
	docker pull cnstark/pytorch:2.4.1-py3.10.15-cuda12.1.0-ubuntu22.04
	docker pull cnstark/pytorch:2.4.1-py3.10.15-cuda12.1.0-devel-ubuntu22.04
	docker pull cnstark/pytorch:2.4.1-py3.10.15-ubuntu22.04

2.4.0

Image	Pull Command
	docker pull cnstark/pytorch:2.4.0-py3.10.15-cuda12.1.0-ubuntu22.04
	docker pull cnstark/pytorch:2.4.0-py3.10.15-cuda12.1.0-devel-ubuntu22.04
	docker pull cnstark/pytorch:2.4.0-py3.10.15-ubuntu22.04

2.3.1

Image	Pull Command
	docker pull cnstark/pytorch:2.3.1-py3.10.15-cuda12.1.0-ubuntu22.04
	docker pull cnstark/pytorch:2.3.1-py3.10.15-cuda12.1.0-devel-ubuntu22.04
	docker pull cnstark/pytorch:2.3.1-py3.10.15-ubuntu22.04

2.3.0

Image	Pull Command
	docker pull cnstark/pytorch:2.3.0-py3.10.15-cuda12.1.0-ubuntu22.04
	docker pull cnstark/pytorch:2.3.0-py3.10.15-cuda12.1.0-devel-ubuntu22.04
	docker pull cnstark/pytorch:2.3.0-py3.10.15-ubuntu22.04

2.2.2

Image	Pull Command
	docker pull cnstark/pytorch:2.2.2-py3.10.15-cuda12.1.0-ubuntu22.04
	docker pull cnstark/pytorch:2.2.2-py3.10.15-cuda12.1.0-devel-ubuntu22.04
	docker pull cnstark/pytorch:2.2.2-py3.10.15-ubuntu22.04

2.2.1

Image	Pull Command
	docker pull cnstark/pytorch:2.2.1-py3.10.15-cuda12.1.0-ubuntu22.04
	docker pull cnstark/pytorch:2.2.1-py3.10.15-cuda12.1.0-devel-ubuntu22.04
	docker pull cnstark/pytorch:2.2.1-py3.10.15-ubuntu22.04

2.2.0

Image	Pull Command
	docker pull cnstark/pytorch:2.2.0-py3.10.15-cuda12.1.0-ubuntu22.04
	docker pull cnstark/pytorch:2.2.0-py3.10.15-cuda12.1.0-devel-ubuntu22.04
	docker pull cnstark/pytorch:2.2.0-py3.10.15-ubuntu22.04

2.1.2

Image	Pull Command
	docker pull cnstark/pytorch:2.1.2-py3.10.15-cuda12.1.0-ubuntu22.04
	docker pull cnstark/pytorch:2.1.2-py3.10.15-cuda12.1.0-devel-ubuntu22.04
	docker pull cnstark/pytorch:2.1.2-py3.10.15-ubuntu22.04

2.1.1

Image	Pull Command
	docker pull cnstark/pytorch:2.1.1-py3.10.15-cuda12.1.0-ubuntu22.04
	docker pull cnstark/pytorch:2.1.1-py3.10.15-cuda12.1.0-devel-ubuntu22.04
	docker pull cnstark/pytorch:2.1.1-py3.10.15-ubuntu22.04

2.1.0

Image	Pull Command
	docker pull cnstark/pytorch:2.1.0-py3.10.15-cuda12.1.0-ubuntu22.04
	docker pull cnstark/pytorch:2.1.0-py3.10.15-cuda12.1.0-devel-ubuntu22.04
	docker pull cnstark/pytorch:2.1.0-py3.10.15-ubuntu22.04

2.0.1

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

2.0.0

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

1.13.1

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

1.13.0

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

1.12.1

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

1.12.0

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

1.11.0

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

1.10.2

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

1.10.1

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

1.10.0

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

1.9.1

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

1.9.0

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

1.8.1

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

1.8.0

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

1.7.1

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

1.7.0

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

1.6.0

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

1.5.1

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

1.5.0

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

1.4.0

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

1.2.0

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

Other Images

Generate Build Script

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

scripts/build_xxx.sh

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

Available Versions

OS Versions

OS	OS version
Ubuntu	22.04, 20.04, 18.04, 16.04, 14.04
CentOS	8, 7, 6

CUDA Versions

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)

Pytorch Versions

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