Pytorch Docker Images

English | 简体中文

Note

This is a fork from cnstark/pytorch-docker supporting almost all combinations of pytorch 1.4-2.5, python 3.8-3.12, ubuntu 18.04-22.04, centos 8, and cuda 10.0-12.4.

Pure pytorch docker images with different os, cuda, and python versions.

Github: https://github.com/xiongsp/pytorch-docker

Docker Hub: https://hub.docker.com/r/spxiong/pytorch

Changelog

• 2024/12/01: Support for pytorch 1.4-2.0.
• 2024/11/05: Support for pytorch 2.5.x. For 2.1-, please refer to this DockerHub and Github Repo.
• 2024/10/10: Support for pytorch 2.1.x 2.2.x 2.3.x 2.4.x (I'm back).

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 \
    spxiong/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

2.5.1

Image	Pull Command
	docker pull spxiong/pytorch:2.5.1-py3.10.15-ubuntu22.04
	docker pull spxiong/pytorch:2.5.1-py3.10.15-cuda12.1.0-ubuntu22.04
	docker pull spxiong/pytorch:2.5.1-py3.10.15-cuda12.1.0-devel-ubuntu22.04
	docker pull spxiong/pytorch:2.5.1-py3.10.15-ubuntu22.04
	docker pull spxiong/pytorch:2.5.1-py3.10.15-cuda12.4.1-ubuntu22.04
	docker pull spxiong/pytorch:2.5.1-py3.10.15-cuda12.4.1-devel-ubuntu22.04
	docker pull spxiong/pytorch:2.5.1-py3.11.10-ubuntu22.04
	docker pull spxiong/pytorch:2.5.1-py3.11.10-cuda12.1.0-ubuntu22.04
	docker pull spxiong/pytorch:2.5.1-py3.11.10-cuda12.1.0-devel-ubuntu22.04
	docker pull spxiong/pytorch:2.5.1-py3.11.10-ubuntu22.04
	docker pull spxiong/pytorch:2.5.1-py3.11.10-cuda12.4.1-ubuntu22.04
	docker pull spxiong/pytorch:2.5.1-py3.11.10-cuda12.4.1-devel-ubuntu22.04
	docker pull spxiong/pytorch:2.5.1-py3.12.7-ubuntu22.04
	docker pull spxiong/pytorch:2.5.1-py3.12.7-cuda12.1.0-ubuntu22.04
	docker pull spxiong/pytorch:2.5.1-py3.12.7-cuda12.1.0-devel-ubuntu22.04
	docker pull spxiong/pytorch:2.5.1-py3.12.7-ubuntu22.04
	docker pull spxiong/pytorch:2.5.1-py3.12.7-cuda12.4.1-ubuntu22.04
	docker pull spxiong/pytorch:2.5.1-py3.12.7-cuda12.4.1-devel-ubuntu22.04

2.5.0

Image	Pull Command
	docker pull spxiong/pytorch:2.5.0-py3.10.15-ubuntu22.04
	docker pull spxiong/pytorch:2.5.0-py3.10.15-cuda12.1.0-ubuntu22.04
	docker pull spxiong/pytorch:2.5.0-py3.10.15-cuda12.1.0-devel-ubuntu22.04
	docker pull spxiong/pytorch:2.5.0-py3.10.15-ubuntu22.04
	docker pull spxiong/pytorch:2.5.0-py3.10.15-cuda12.4.1-ubuntu22.04
	docker pull spxiong/pytorch:2.5.0-py3.10.15-cuda12.4.1-devel-ubuntu22.04
	docker pull spxiong/pytorch:2.5.0-py3.11.10-ubuntu22.04
	docker pull spxiong/pytorch:2.5.0-py3.11.10-cuda12.1.0-ubuntu22.04
	docker pull spxiong/pytorch:2.5.0-py3.11.10-cuda12.1.0-devel-ubuntu22.04
	docker pull spxiong/pytorch:2.5.0-py3.11.10-ubuntu22.04
	docker pull spxiong/pytorch:2.5.0-py3.11.10-cuda12.4.1-ubuntu22.04
	docker pull spxiong/pytorch:2.5.0-py3.11.10-cuda12.4.1-devel-ubuntu22.04
	docker pull spxiong/pytorch:2.5.0-py3.12.7-ubuntu22.04
	docker pull spxiong/pytorch:2.5.0-py3.12.7-cuda12.1.0-ubuntu22.04
	docker pull spxiong/pytorch:2.5.0-py3.12.7-cuda12.1.0-devel-ubuntu22.04
	docker pull spxiong/pytorch:2.5.0-py3.12.7-ubuntu22.04
	docker pull spxiong/pytorch:2.5.0-py3.12.7-cuda12.4.1-ubuntu22.04
	docker pull spxiong/pytorch:2.5.0-py3.12.7-cuda12.4.1-devel-ubuntu22.04

2.1-2.4

2.4.1

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

2.4.0

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

2.3.1

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

2.3.0

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

2.2.2

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

2.2.1

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

2.2.0

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

2.1.2

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

2.1.1

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

2.1.0

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

<= 2.0

2.0.1

Image	Pull Command
	docker pull spxiong/pytorch:2.0.1-py3.9.12-ubuntu22.04
	docker pull spxiong/pytorch:2.0.1-py3.9.12-cuda11.7.1-ubuntu22.04
	docker pull spxiong/pytorch:2.0.1-py3.9.12-cuda11.7.1-devel-ubuntu22.04
	docker pull spxiong/pytorch:2.0.1-py3.9.12-cuda11.8.0-ubuntu22.04
	docker pull spxiong/pytorch:2.0.1-py3.9.12-cuda11.8.0-devel-ubuntu22.04

2.0.0

Image	Pull Command
	docker pull spxiong/pytorch:2.0.0-py3.9.12-ubuntu22.04
	docker pull spxiong/pytorch:2.0.0-py3.9.12-cuda11.7.1-ubuntu22.04
	docker pull spxiong/pytorch:2.0.0-py3.9.12-cuda11.7.1-devel-ubuntu22.04
	docker pull spxiong/pytorch:2.0.0-py3.9.12-cuda11.8.0-ubuntu22.04
	docker pull spxiong/pytorch:2.0.0-py3.9.12-cuda11.8.0-devel-ubuntu22.04

1.13.1

Image	Pull Command
	docker pull spxiong/pytorch:1.13.1-py3.8.13-ubuntu20.04
	docker pull spxiong/pytorch:1.13.1-py3.8.13-cuda11.6.2-ubuntu20.04
	docker pull spxiong/pytorch:1.13.1-py3.8.13-cuda11.6.2-devel-ubuntu20.04
	docker pull spxiong/pytorch:1.13.1-py3.8.13-cuda11.7.1-ubuntu20.04
	docker pull spxiong/pytorch:1.13.1-py3.8.13-cuda11.7.1-devel-ubuntu20.04
	docker pull spxiong/pytorch:1.13.1-py3.9.12-ubuntu20.04
	docker pull spxiong/pytorch:1.13.1-py3.9.12-cuda11.6.2-ubuntu20.04
	docker pull spxiong/pytorch:1.13.1-py3.9.12-cuda11.6.2-devel-ubuntu20.04
	docker pull spxiong/pytorch:1.13.1-py3.9.12-cuda11.7.1-ubuntu20.04
	docker pull spxiong/pytorch:1.13.1-py3.9.12-cuda11.7.1-devel-ubuntu20.04

1.13.0

Image	Pull Command
	docker pull spxiong/pytorch:1.13.0-py3.8.13-ubuntu20.04
	docker pull spxiong/pytorch:1.13.0-py3.8.13-cuda11.6.2-ubuntu20.04
	docker pull spxiong/pytorch:1.13.0-py3.8.13-cuda11.6.2-devel-ubuntu20.04
	docker pull spxiong/pytorch:1.13.0-py3.8.13-cuda11.7.1-ubuntu20.04
	docker pull spxiong/pytorch:1.13.0-py3.8.13-cuda11.7.1-devel-ubuntu20.04
	docker pull spxiong/pytorch:1.13.0-py3.9.12-ubuntu20.04
	docker pull spxiong/pytorch:1.13.0-py3.9.12-cuda11.6.2-ubuntu20.04
	docker pull spxiong/pytorch:1.13.0-py3.9.12-cuda11.6.2-devel-ubuntu20.04
	docker pull spxiong/pytorch:1.13.0-py3.9.12-cuda11.7.1-ubuntu20.04
	docker pull spxiong/pytorch:1.13.0-py3.9.12-cuda11.7.1-devel-ubuntu20.04

1.12.1

Image	Pull Command
	docker pull spxiong/pytorch:1.12.1-py3.8.13-ubuntu20.04
	docker pull spxiong/pytorch:1.12.1-py3.8.13-cuda11.3.1-ubuntu20.04
	docker pull spxiong/pytorch:1.12.1-py3.8.13-cuda11.3.1-devel-ubuntu20.04
	docker pull spxiong/pytorch:1.12.1-py3.8.13-cuda11.6.2-ubuntu20.04
	docker pull spxiong/pytorch:1.12.1-py3.8.13-cuda11.6.2-devel-ubuntu20.04
	docker pull spxiong/pytorch:1.12.1-py3.9.12-ubuntu20.04
	docker pull spxiong/pytorch:1.12.1-py3.9.12-cuda11.3.1-ubuntu20.04
	docker pull spxiong/pytorch:1.12.1-py3.9.12-cuda11.3.1-devel-ubuntu20.04
	docker pull spxiong/pytorch:1.12.1-py3.9.12-cuda11.6.2-ubuntu20.04
	docker pull spxiong/pytorch:1.12.1-py3.9.12-cuda11.6.2-devel-ubuntu20.04

1.12.0

Image	Pull Command
	docker pull spxiong/pytorch:1.12.0-py3.8.13-ubuntu20.04
	docker pull spxiong/pytorch:1.12.0-py3.8.13-cuda11.3.1-ubuntu20.04
	docker pull spxiong/pytorch:1.12.0-py3.8.13-cuda11.3.1-devel-ubuntu20.04
	docker pull spxiong/pytorch:1.12.0-py3.8.13-cuda11.6.2-ubuntu20.04
	docker pull spxiong/pytorch:1.12.0-py3.8.13-cuda11.6.2-devel-ubuntu20.04
	docker pull spxiong/pytorch:1.12.0-py3.9.12-ubuntu20.04
	docker pull spxiong/pytorch:1.12.0-py3.9.12-cuda11.3.1-ubuntu20.04
	docker pull spxiong/pytorch:1.12.0-py3.9.12-cuda11.3.1-devel-ubuntu20.04
	docker pull spxiong/pytorch:1.12.0-py3.9.12-cuda11.6.2-ubuntu20.04
	docker pull spxiong/pytorch:1.12.0-py3.9.12-cuda11.6.2-devel-ubuntu20.04

1.11.0

Image	Pull Command
	docker pull spxiong/pytorch:1.11.0-py3.8.13-ubuntu20.04
	docker pull spxiong/pytorch:1.11.0-py3.8.13-cuda11.3.1-ubuntu20.04
	docker pull spxiong/pytorch:1.11.0-py3.8.13-cuda11.3.1-devel-ubuntu20.04
	docker pull spxiong/pytorch:1.11.0-py3.9.12-ubuntu20.04
	docker pull spxiong/pytorch:1.11.0-py3.9.12-cuda11.3.1-ubuntu20.04
	docker pull spxiong/pytorch:1.11.0-py3.9.12-cuda11.3.1-devel-ubuntu20.04

1.10.2

Image	Pull Command
	docker pull spxiong/pytorch:1.10.2-py3.8.13-ubuntu20.04
	docker pull spxiong/pytorch:1.10.2-py3.8.13-cuda11.3.1-ubuntu20.04
	docker pull spxiong/pytorch:1.10.2-py3.8.13-cuda11.3.1-devel-ubuntu20.04
	docker pull spxiong/pytorch:1.10.2-py3.9.12-ubuntu20.04
	docker pull spxiong/pytorch:1.10.2-py3.9.12-cuda11.3.1-ubuntu20.04
	docker pull spxiong/pytorch:1.10.2-py3.9.12-cuda11.3.1-devel-ubuntu20.04

1.10.1

Image	Pull Command
	docker pull spxiong/pytorch:1.10.1-py3.8.13-ubuntu20.04
	docker pull spxiong/pytorch:1.10.1-py3.8.13-cuda11.1.1-ubuntu20.04
	docker pull spxiong/pytorch:1.10.1-py3.8.13-cuda11.1.1-devel-ubuntu20.04
	docker pull spxiong/pytorch:1.10.1-py3.9.12-ubuntu20.04
	docker pull spxiong/pytorch:1.10.1-py3.9.12-cuda11.1.1-ubuntu20.04
	docker pull spxiong/pytorch:1.10.1-py3.9.12-cuda11.1.1-devel-ubuntu20.04

1.10.0

Image	Pull Command
	docker pull spxiong/pytorch:1.10.0-py3.8.13-ubuntu20.04
	docker pull spxiong/pytorch:1.10.0-py3.8.13-cuda11.1.1-ubuntu20.04
	docker pull spxiong/pytorch:1.10.0-py3.8.13-cuda11.1.1-devel-ubuntu20.04
	docker pull spxiong/pytorch:1.10.0-py3.9.12-ubuntu20.04
	docker pull spxiong/pytorch:1.10.0-py3.9.12-cuda11.1.1-ubuntu20.04
	docker pull spxiong/pytorch:1.10.0-py3.9.12-cuda11.1.1-devel-ubuntu20.04

1.9.1

Image	Pull Command
	docker pull spxiong/pytorch:1.9.1-py3.8.13-ubuntu20.04
	docker pull spxiong/pytorch:1.9.1-py3.8.13-cuda11.1.1-ubuntu20.04
	docker pull spxiong/pytorch:1.9.1-py3.8.13-cuda11.1.1-devel-ubuntu20.04
	docker pull spxiong/pytorch:1.9.1-py3.9.12-ubuntu20.04
	docker pull spxiong/pytorch:1.9.1-py3.9.12-cuda11.1.1-ubuntu20.04
	docker pull spxiong/pytorch:1.9.1-py3.9.12-cuda11.1.1-devel-ubuntu20.04

1.9.0

Image	Pull Command
	docker pull spxiong/pytorch:1.9.0-py3.8.13-ubuntu20.04
	docker pull spxiong/pytorch:1.9.0-py3.8.13-cuda11.1.1-ubuntu20.04
	docker pull spxiong/pytorch:1.9.0-py3.8.13-cuda11.1.1-devel-ubuntu20.04
	docker pull spxiong/pytorch:1.9.0-py3.9.12-ubuntu20.04
	docker pull spxiong/pytorch:1.9.0-py3.9.12-cuda11.1.1-ubuntu20.04
	docker pull spxiong/pytorch:1.9.0-py3.9.12-cuda11.1.1-devel-ubuntu20.04

1.8.1

Image	Pull Command
	docker pull spxiong/pytorch:1.8.1-py3.8.13-ubuntu20.04
	docker pull spxiong/pytorch:1.8.1-py3.8.13-cuda11.1.1-ubuntu20.04
	docker pull spxiong/pytorch:1.8.1-py3.8.13-cuda11.1.1-devel-ubuntu20.04
	docker pull spxiong/pytorch:1.8.1-py3.9.12-ubuntu20.04
	docker pull spxiong/pytorch:1.8.1-py3.9.12-cuda11.1.1-ubuntu20.04
	docker pull spxiong/pytorch:1.8.1-py3.9.12-cuda11.1.1-devel-ubuntu20.04

1.8.0

Image	Pull Command
	docker pull spxiong/pytorch:1.8.0-py3.8.13-ubuntu20.04
	docker pull spxiong/pytorch:1.8.0-py3.8.13-cuda11.1.1-ubuntu20.04
	docker pull spxiong/pytorch:1.8.0-py3.8.13-cuda11.1.1-devel-ubuntu20.04
	docker pull spxiong/pytorch:1.8.0-py3.9.12-ubuntu20.04
	docker pull spxiong/pytorch:1.8.0-py3.9.12-cuda11.1.1-ubuntu20.04
	docker pull spxiong/pytorch:1.8.0-py3.9.12-cuda11.1.1-devel-ubuntu20.04

1.7.1

Image	Pull Command
	docker pull spxiong/pytorch:1.7.1-py3.8.13-ubuntu18.04
	docker pull spxiong/pytorch:1.7.1-py3.8.13-cuda10.1-ubuntu18.04
	docker pull spxiong/pytorch:1.7.1-py3.8.13-cuda10.1-devel-ubuntu18.04
	docker pull spxiong/pytorch:1.7.1-py3.8.13-cuda10.2-ubuntu18.04
	docker pull spxiong/pytorch:1.7.1-py3.8.13-cuda10.2-devel-ubuntu18.04
	docker pull spxiong/pytorch:1.7.1-py3.8.13-cuda11.0.3-ubuntu18.04
	docker pull spxiong/pytorch:1.7.1-py3.8.13-cuda11.0.3-devel-ubuntu18.04
	docker pull spxiong/pytorch:1.7.1-py3.9.12-ubuntu18.04
	docker pull spxiong/pytorch:1.7.1-py3.9.12-cuda10.1-ubuntu18.04
	docker pull spxiong/pytorch:1.7.1-py3.9.12-cuda10.1-devel-ubuntu18.04
	docker pull spxiong/pytorch:1.7.1-py3.9.12-cuda10.2-ubuntu18.04
	docker pull spxiong/pytorch:1.7.1-py3.9.12-cuda10.2-devel-ubuntu18.04
	docker pull spxiong/pytorch:1.7.1-py3.9.12-cuda11.0.3-ubuntu18.04
	docker pull spxiong/pytorch:1.7.1-py3.9.12-cuda11.0.3-devel-ubuntu18.04

1.7.0

Image	Pull Command
	docker pull spxiong/pytorch:1.7.0-py3.8.13-ubuntu18.04
	docker pull spxiong/pytorch:1.7.0-py3.8.13-cuda10.1-ubuntu18.04
	docker pull spxiong/pytorch:1.7.0-py3.8.13-cuda10.1-devel-ubuntu18.04
	docker pull spxiong/pytorch:1.7.0-py3.8.13-cuda10.2-ubuntu18.04
	docker pull spxiong/pytorch:1.7.0-py3.8.13-cuda10.2-devel-ubuntu18.04
	docker pull spxiong/pytorch:1.7.0-py3.8.13-cuda11.0.3-ubuntu18.04
	docker pull spxiong/pytorch:1.7.0-py3.8.13-cuda11.0.3-devel-ubuntu18.04

1.6.0

Image	Pull Command
	docker pull spxiong/pytorch:1.6.0-py3.8.13-ubuntu18.04
	docker pull spxiong/pytorch:1.6.0-py3.8.13-cuda10.1-ubuntu18.04
	docker pull spxiong/pytorch:1.6.0-py3.8.13-cuda10.1-devel-ubuntu18.04
	docker pull spxiong/pytorch:1.6.0-py3.8.13-cuda10.2-ubuntu18.04
	docker pull spxiong/pytorch:1.6.0-py3.8.13-cuda10.2-devel-ubuntu18.04

1.5.1

Image	Pull Command
	docker pull spxiong/pytorch:1.5.1-py3.8.13-ubuntu18.04
	docker pull spxiong/pytorch:1.5.1-py3.8.13-cuda10.1-ubuntu18.04
	docker pull spxiong/pytorch:1.5.1-py3.8.13-cuda10.1-devel-ubuntu18.04
	docker pull spxiong/pytorch:1.5.1-py3.8.13-cuda10.2-ubuntu18.04
	docker pull spxiong/pytorch:1.5.1-py3.8.13-cuda10.2-devel-ubuntu18.04

1.5.0

Image	Pull Command
	docker pull spxiong/pytorch:1.5.0-py3.8.13-ubuntu18.04
	docker pull spxiong/pytorch:1.5.0-py3.8.13-cuda10.1-ubuntu18.04
	docker pull spxiong/pytorch:1.5.0-py3.8.13-cuda10.1-devel-ubuntu18.04
	docker pull spxiong/pytorch:1.5.0-py3.8.13-cuda10.2-ubuntu18.04
	docker pull spxiong/pytorch:1.5.0-py3.8.13-cuda10.2-devel-ubuntu18.04

1.4.0

Image	Pull Command
	docker pull spxiong/pytorch:1.4.0-py3.8.13-ubuntu18.04
	docker pull spxiong/pytorch:1.4.0-py3.8.13-cuda10.1-ubuntu18.04
	docker pull spxiong/pytorch:1.4.0-py3.8.13-cuda10.1-devel-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