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