Docker Hub images

Here are images available from Docker Hub presence.

Click on one of the domain buttons to see the relevant frameworks and images:

PyTorch (using our wai.pytorchimageclass Python library)
- PyTorch 1.6.0, CUDA 10.1
  
  waikatodatamining/pytorch-image-classification:1.6
- PyTorch 1.6.0, CPU
  
  waikatodatamining/pytorch-image-classification:1.6_cpu
PyTorch/MMClassification
- MMClassification 0.25.0, PyTorch 1.9 and CUDA 11.1
  
  waikatodatamining/mmclassification:0.25.0_cuda11.1
- MMClassification 0.25.0, PyTorch 1.9 and CPU
  
  waikatodatamining/mmclassification:0.25.0_cpu
PyTorch/MMPretrain (successor of MMClassification)
- MMPretrain 1.2.0, PyTorch 1.9 and CUDA 11.1
  
  waikatodatamining/mmpretrain:1.2.0_cuda11.1
- MMPretrain 1.2.0, PyTorch 1.11 and CPU
  
  waikatodatamining/mmpretrain:1.2.0_cpu
PyTorch/PaddleClas
- PaddleClas 2.6.0, PyTorch 2.5.1 and CUDA 11.8
  
  waikatodatamining/paddleclas:2.6.0_cuda11.8
- PaddleClas 2.6.0, PyTorch 1.11.0 and CPU
  
  waikatodatamining/paddleclas:2.6.0_cpu
Tensorflow (using our wai.tfimageclass Python library)
- Tensorflow 1.14, CUDA 10.0
  
  waikatodatamining/tf_image_classification:1.14
- Tensorflow 1.14, CPU
  
  waikatodatamining/tf_image_classification:1.14_cpu
Tensorflow (using our wai.tflite_model_maker Python library)
- 2.4.3, CUDA 11.0
  
  waikatodatamining/tflite_model_maker:2.4.3
- 2.4.3, CPU
  
  waikatodatamining/tflite_model_maker:2.4.3_cpu
Tensorflow (using the make_image_classifier Python library)
- 2.9.1, CUDA 11.1
  
  waikatodatamining/tf_image_classification2:2.9.1_cuda11.1
- 2.9.1, CPU
  
  waikatodatamining/tf_image_classification2:2.9.1_cpu

PyTorch/MMDetection
- MMDetection 3.1.0, PyTorch 1.12 and CUDA 11.3
  
  waikatodatamining/mmdetection:3.1.0_cuda11.3
- MMDetection 3.1.0, PyTorch 1.12 and CPU
  
  waikatodatamining/mmdetection:3.1.0_cpu
Tensorflow (using our wai.tflite_model_maker Python library)
- 2.4.3, CUDA 11.0
  
  waikatodatamining/tflite_model_maker:2.4.3
- 2.4.3, CPU
  
  waikatodatamining/tflite_model_maker:2.4.3_cpu
PyTorch/PaddleDetection
- PaddleDetection 2.8.0, CPU
  
  waikatodatamining/paddledetection:2.8.0_cpu
- PaddleDetection 2.8.0, CUDA 11.8
  
  waikatodatamining/paddledetection:2.8.0_cuda11.8
PyTorch/Yolov5
- Yolov5 2022-11-05, CPU
  
  waikatodatamining/pytorch-yolov5:2022-11-05_cpu
- Yolov5 2022-11-05, CUDA 11.1
  
  waikatodatamining/pytorch-yolov5:2022-11-05_cuda11.1
PyTorch/Yolov7
- Yolov7 2022-10-08, CPU
  
  waikatodatamining/pytorch-yolov7:2022-10-08_cpu
- Yolov7 2022-10-08, CUDA 11.1
  
  waikatodatamining/pytorch-yolov7:2022-10-08_cuda11.1
PyTorch/Yolov10
- Yolov10 2024-06-23, CPU
  
  waikatodatamining/pytorch-yolov10:2024-06-23_cpu
- Yolov10 2024-06-23, CUDA 11.7
  
  waikatodatamining/pytorch-yolov10:2024-06-23_cuda11.1
PyTorch/Yolo26
- Yolo26 8.4.16, CPU
  
  waikatodatamining/pytorch-yolo26:8.4.16_cpu
- Yolo26 8.4.16, CUDA 11.7
  
  waikatodatamining/pytorch-yolo26:8.4.16_cuda12.6

Tensorflow/Keras
- image-segmentation-keras 0.3.0, Tensorflow 1.14, CUDA 10.0
  
  waikatodatamining/image-segmentation-keras:1.14.0_0.3.0
- image-segmentation-keras 0.3.0, Tensorflow 2.4.1, CUDA 11.0
  
  waikatodatamining/image-segmentation-keras:2.4.1_0.3.0
PyTorch
- MMSegmentation 1.2.2, PyTorch 1.11.0 and CUDA 11.3
  
  waikatodatamining/mmsegmentation:1.2.2-3_cuda11.3
- MMSegmentation 1.2.2, PyTorch 1.11.0 and CPU
  
  waikatodatamining/mmsegmentation:1.2.2-3_cpu
- PaddleSeg 2.10.0, PyTorch 2.7.0 and CUDA 11.8
  
  waikatodatamining/paddleseg:2.10.0-3_cuda11.8
- PaddleSeg 2.10.0, PyTorch 2.7.0 and CPU
  
  waikatodatamining/paddleseg:2.10.0-3_cpu
- DEXTR 0.1.2, PyTorch 1.10.0 and CUDA 11.1
  
  waikatodatamining/pytorch-dextr:0.1.2_cuda11.1
- DEXTR 0.1.2, PyTorch 1.9.1 and CPU
  
  waikatodatamining/pytorch-dextr:0.1.2_cpu
- Segment-Anything Model 2023-04-16, PyTorch 1.13.0 and CUDA 11.6
  
  waikatodatamining/pytorch-sam:2023-04-16_cuda11.6
- Segment-Anything Model 2023-04-16, PyTorch 1.9.1 and CPU
  
  waikatodatamining/pytorch-sam:2023-04-16_cpu
- Segment-Anything in High Quality 2023-08-17, PyTorch 1.13.0 and CUDA 11.6
  
  waikatodatamining/pytorch-sam-hq:2023-08-17_cuda11.6
- Segment-Anything in High Quality 2023-08-17, PyTorch 1.9.1 and CPU
  
  waikatodatamining/pytorch-sam-hq:2023-08-17_cpu

PyTorch/Detectron2
- Detectron2 0.5, PyTorch 1.9, CUDA 11.1
  
  waikatodatamining/detectron2:0.5
- Detectron2 0.6, PyTorch 1.9, CUDA 11.1
  
  waikatodatamining/detectron2:0.6
PyTorch/YOLACT++
- YOLACT++ 2020-02-11, PyTorch 1.2, CUDA 10.0
  
  waikatodatamining/yolactpp:2020-02-11
PyTorch/Yolov5
- Yolov5 2022-11-05, CPU
  
  waikatodatamining/pytorch-yolov5:2022-11-05_cpu
- Yolov5 2022-11-05, CUDA 11.1
  
  waikatodatamining/pytorch-yolov5:2022-11-05_cuda11.1

BitNet
- BitNet 2025-04-30, CPU
  
  waikatodatamining/bitnet:2025-05-30_cpu
Falcontune
- Falcontune 20230618, CUDA 11.7
  
  waikatodatamining/pytorch-huggingface-transformers:4.31.0_cuda11.7_falcontune_20230618
finetune-gpt2xl
- finetune-gpt2xl 20220924, CUDA 11.1
  
  waikatodatamining/pytorch-huggingface-transformers:4.7.0_cuda11.1_finetune-gpt2xl_20220924
Llama-2
- Llama2, CUDA 11.7
  
  waikatodatamining/pytorch-huggingface-transformers:4.31.0_cuda11.7_llama2
XTuner
- XTuner 2024-02-19, CUDA 11.7
  
  waikatodatamining/pytorch-xtuner:2024-02-19_cuda11.7
- XTuner 0.1.15, CUDA 11.7
  
  waikatodatamining/pytorch-xtuner:0.1.15_cuda11.7
- XTuner 0.1.18, CUDA 11.7
  
  waikatodatamining/pytorch-xtuner:0.1.18_cuda11.7

OpenAI Whisper
- Faster Whisper 1.0.2, CUDA 12.1
  
  waikatodatamining/python-faster-whisper:1.0.2_cuda12.1
- Faster Whisper 1.0.2, CPU
  
  waikatodatamining/python-faster-whisper:1.0.2_cpu

Huggingface
- Huggingface 4.36.0, CUDA 11.7
  
  waikatodatamining/pytorch-huggingface-transformers:4.36.0_cuda11.7_classification

For some of the frameworks, we have tutorials available on how to use them in practice. This includes data preparation, training and making predictions. Check out our Applied Deep Learning site for more details.