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

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.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

pytorch-docker
pytorch-docker copied to clipboard

Metadata

Pytorch Docker Images

Changelog

Usage

Preparation

Use pytorch docker

Image List (More images are on the way ~)

2.0.1

2.0.0

1.13.1

1.13.0

1.12.1

1.12.0

1.11.0

1.10.2

1.10.1

1.10.0

1.9.1

1.9.0

1.8.1

1.8.0

1.7.1

1.7.0

1.6.0

1.5.1

1.5.0

1.4.0

1.2.0

Other Images

Generate Build Script

Build Pytorch Docker Image

Commit the Version (Optional)

Available Versions

OS Versions

CUDA Versions

Pytorch Versions

← Metadata

Owner

Metadata

pytorch-docker pytorch-docker copied to clipboard

Metadata

Pytorch Docker Images

Changelog

Usage

Preparation

Use pytorch docker

Image List (More images are on the way ~)

2.0.1

2.0.0

1.13.1

1.13.0

1.12.1

1.12.0

1.11.0

1.10.2

1.10.1

1.10.0

1.9.1

1.9.0

1.8.1

1.8.0

1.7.1

1.7.0

1.6.0

1.5.1

1.5.0

1.4.0

1.2.0

Other Images

Generate Build Script

Build Pytorch Docker Image

Commit the Version (Optional)

Available Versions

OS Versions

CUDA Versions

Pytorch Versions

← Metadata

Owner

Metadata

pytorch-docker
pytorch-docker copied to clipboard