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
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]
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.
scripts/build_xxx.sh
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 | 20.04, 18.04, 16.04, 14.04 |
CentOS | 8, 7, 6 |
CUDA | CuDNN | OS(version) |
---|---|---|
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 |
---|---|
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 |