A collection of some Docker images for machine learning
| Dockerfile | Docker Hub | CUDA | cuDNN | OS | Python | Torch | Tensorflow | Extra | Size |
|---|---|---|---|---|---|---|---|---|---|
| Dockerfile | Tag | 12.1.0 | 8 | Ubuntu 22.04 | 3.11 | 2.2.0 | 2.15.0 | transformers + diffusers + accelerate | 13.18 GB |
| Dockerfile | Tag | 11.8.0 | 8 | Ubuntu 22.04 | 3.11 | 2.0.1 | 2.13.0 | - | 11.53 GB |
| Dockerfile | Tag | 11.7.1 | 8 | Ubuntu 20.04 | 3.10 | 1.13.1 | 2.11.0 | Detectron2 | 9.53 GB |
| Dockerfile | Tag | - | - | Ubuntu 20.04 | 3.8 | 1.11.0 | 2.9.0 | Detectron2 | 3.48 GB |