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
- Docker
- Nvidia-docker (If a CUDA is required)
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 | Pull Command |
|---|---|
docker pull cnstark/pytorch:1.13.1-py3.9.12-cuda11.7.1-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.9.12-ubuntu20.04 |
|
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 |
|
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 |
|
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-ubuntu20.04 |
|
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 |
|
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 |
|
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 |
|
docker pull cnstark/pytorch:1.10.0-py3.9.12-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.9.12-ubuntu20.04 |
|
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 |
|
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 |
|
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 |
|
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 |
|
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 |
|
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 |
|
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 |
|
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 |
|
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 |
|
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 |
|
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 |
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 imagescripts/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 |