In-house Docker images

In order to access these images from our in-house registry, you need to log in using the following credentials:

docker login -u public -p public public.aml-repo.cms.waikato.ac.nz:443

When pulling images, you need to ensure that you are using the following prefix (the pull commands do not show this, due to using a reverse proxy):

public.aml-repo.cms.waikato.ac.nz:443/

For some of the frameworks, we have tutorials available on how to use them in practice. This includes data preparation, training and making predictions.

https://www.data-mining.co.nz/applied-deep-learning/

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
  
  public.aml-repo.cms.waikato.ac.nz:443/pytorch/pytorch-image-classification:1.6
- PyTorch 1.6.0, CPU
  
  public.aml-repo.cms.waikato.ac.nz:443/pytorch/pytorch-image-classification:1.6_cpu
PyTorch/MMClassification
- MMClassification 0.23.1, PyTorch 1.9 and CUDA 11.1
  
  public.aml-repo.cms.waikato.ac.nz:443/open-mmlab/mmclassification:0.23.1_cuda11.1
- MMClassification 0.23.1, PyTorch 1.9 and CPU
  
  public.aml-repo.cms.waikato.ac.nz:443/open-mmlab/mmclassification:0.23.1_cpu
- MMClassification 0.25.0, PyTorch 1.9 and CUDA 11.1
  
  public.aml-repo.cms.waikato.ac.nz:443/open-mmlab/mmclassification:0.25.0_cuda11.1
- MMClassification 0.25.0, PyTorch 1.9 and CPU
  
  public.aml-repo.cms.waikato.ac.nz:443/open-mmlab/mmclassification:0.25.0_cpu
PyTorch/MMPretrain (successor of MMClassification)
- MMPretrain 1.2.0, PyTorch 1.9 and CUDA 11.1
  
  public.aml-repo.cms.waikato.ac.nz:443/open-mmlab/mmpretrain:1.2.0_cuda11.1
- MMPretrain 1.2.0, PyTorch 1.11 and CPU
  
  public.aml-repo.cms.waikato.ac.nz:443/open-mmlab/mmpretrain:1.2.0_cpu
Tensorflow (using our wai.tfimageclass Python library)
- Tensorflow 1.14, CUDA 10.0
  
  public.aml-repo.cms.waikato.ac.nz:443/tensorflow/tf_image_classification:1.14
- Tensorflow 1.14, CPU
  
  public.aml-repo.cms.waikato.ac.nz:443/tensorflow/tf_image_classification:1.14_cpu
Tensorflow (using our wai.tflite_model_maker Python library)
- 2.4.3, CUDA 11.0
  
  public.aml-repo.cms.waikato.ac.nz:443/tensorflow/tflite_model_maker:2.4.3
- 2.4.3, CPU
  
  public.aml-repo.cms.waikato.ac.nz:443/tensorflow/tflite_model_maker:2.4.3_cpu
Tensorflow (using the make_image_classifier Python library)
- 2.9.1, CUDA 11.1
  
  public.aml-repo.cms.waikato.ac.nz:443/tensorflow/tf_image_classification2:2.9.1_cuda11.1
- 2.9.1, CPU
  
  public.aml-repo.cms.waikato.ac.nz:443/tensorflow/tf_image_classification2:2.9.1_cpu

PyTorch/MMDetection
- MMDetection 2020-03-01, PyTorch 1.3 and CUDA 10.1
  
  public.aml-repo.cms.waikato.ac.nz:443/open-mmlab/mmdetection:2020-03-01
- MMDetection 2.16.0, PyTorch 1.6 and CUDA 10.1
  
  public.aml-repo.cms.waikato.ac.nz:443/open-mmlab/mmdetection:2.16.0
- MMDetection 2.16.0, PyTorch 1.9 and CUDA 11.1
  
  public.aml-repo.cms.waikato.ac.nz:443/open-mmlab/mmdetection:2.16.0_cuda11.1
- MMDetection 2.16.0, PyTorch 1.9 and CPU
  
  public.aml-repo.cms.waikato.ac.nz:443/open-mmlab/mmdetection:2.16.0_cpu
- MMDetection 2.18.1, PyTorch 1.9 and CUDA 11.1
  
  public.aml-repo.cms.waikato.ac.nz:443/open-mmlab/mmdetection:2.18.1_cuda11.1
- MMDetection 2.18.1, PyTorch 1.9 and CPU
  
  public.aml-repo.cms.waikato.ac.nz:443/open-mmlab/mmdetection:2.18.1_cpu
- MMDetection 2.24.1, PyTorch 1.9 and CUDA 11.1
  
  public.aml-repo.cms.waikato.ac.nz:443/open-mmlab/mmdetection:2.24.1_cuda11.1
- MMDetection 2.24.1, PyTorch 1.9 and CPU
  
  public.aml-repo.cms.waikato.ac.nz:443/open-mmlab/mmdetection:2.24.1_cpu
- MMDetection 2.27.0, PyTorch 1.9 and CUDA 11.1
  
  public.aml-repo.cms.waikato.ac.nz:443/open-mmlab/mmdetection:2.27.0_cuda11.1
- MMDetection 2.27.0, PyTorch 1.9 and CPU
  
  public.aml-repo.cms.waikato.ac.nz:443/open-mmlab/mmdetection:2.27.0_cpu
- MMDetection 3.1.0, PyTorch 1.12 and CUDA 11.3
  
  public.aml-repo.cms.waikato.ac.nz:443/open-mmlab/mmdetection:3.1.0_cuda11.3
- MMDetection 3.1.0, PyTorch 1.12 and CPU
  
  public.aml-repo.cms.waikato.ac.nz:443/open-mmlab/mmdetection:3.1.0_cpu
Tensorflow
- Tensorflow 1.14, Object Detection API 2019-08-31, CUDA 10.0
  
  public.aml-repo.cms.waikato.ac.nz:443/tensorflow/tf_object_detection:1.14.0_2019-08-31
Tensorflow (using our wai.tflite_model_maker Python library)
- 2.4.3, CUDA 11.0
  
  public.aml-repo.cms.waikato.ac.nz:443/tensorflow/tflite_model_maker:2.4.3
- 2.4.3, CPU
  
  public.aml-repo.cms.waikato.ac.nz:443/tensorflow/tflite_model_maker:2.4.3_cpu
PyTorch/Yolov5
- Yolov5 2022-01-21, CPU
  
  public.aml-repo.cms.waikato.ac.nz:443/pytorch/pytorch-yolov5:2022-01-21_cpu
- Yolov5 2022-01-21, CUDA 11.1
  
  public.aml-repo.cms.waikato.ac.nz:443/pytorch/pytorch-yolov5:2022-01-21_cuda11.1
- Yolov5 2022-05-31, CPU
  
  public.aml-repo.cms.waikato.ac.nz:443/pytorch/pytorch-yolov5:2022-05-31_cpu
- Yolov5 2022-05-31, CUDA 11.1
  
  public.aml-repo.cms.waikato.ac.nz:443/pytorch/pytorch-yolov5:2022-05-31_cuda11.1
- Yolov5 2022-09-29, CPU
  
  public.aml-repo.cms.waikato.ac.nz:443/pytorch/pytorch-yolov5:2022-09-29_cpu
- Yolov5 2022-09-29, CUDA 11.1
  
  public.aml-repo.cms.waikato.ac.nz:443/pytorch/pytorch-yolov5:2022-09-29_cuda11.1
- Yolov5 2022-11-05, CPU
  
  public.aml-repo.cms.waikato.ac.nz:443/pytorch/pytorch-yolov5:2022-11-05_cpu
- Yolov5 2022-11-05, CUDA 11.1
  
  public.aml-repo.cms.waikato.ac.nz:443/pytorch/pytorch-yolov5:2022-11-05_cuda11.1
PyTorch/Yolov7
- Yolov7 2022-10-08, CPU
  
  public.aml-repo.cms.waikato.ac.nz:443/pytorch/pytorch-yolov7:2022-10-08_cpu
- Yolov7 2022-10-08, CUDA 11.1
  
  public.aml-repo.cms.waikato.ac.nz:443/pytorch/pytorch-yolov7:2022-10-08_cuda11.1

Tensorflow/Keras
- image-segmentation-keras 0.3.0, Tensorflow 1.14, CUDA 10.0
  
  public.aml-repo.cms.waikato.ac.nz:443/tensorflow/image-segmentation-keras:1.14.0_0.3.0
- image-segmentation-keras 0.3.0, Tensorflow 2.4.1, CUDA 11.0
  
  public.aml-repo.cms.waikato.ac.nz:443/tensorflow/image-segmentation-keras:2.4.1_0.3.0
PyTorch
- Segmentation Models 0.2.1, PyTorch 1.9.0, CUDA 11.1
  
  public.aml-repo.cms.waikato.ac.nz:443/pytorch/segmentation_models:0.2.1
- MMSegmentation 0.25.0, PyTorch 1.9 and CUDA 11.1
  
  public.aml-repo.cms.waikato.ac.nz:443/open-mmlab/mmsegmentation:0.25.0_cuda11.1
- MMSegmentation 0.25.0, PyTorch 1.9 and CPU
  
  public.aml-repo.cms.waikato.ac.nz:443/open-mmlab/mmsegmentation:0.25.0_cpu
- MMSegmentation 0.30.0, PyTorch 1.9 and CUDA 11.1
  
  public.aml-repo.cms.waikato.ac.nz:443/open-mmlab/mmsegmentation:0.30.0_cuda11.1
- MMSegmentation 0.30.0, PyTorch 1.9 and CPU
  
  public.aml-repo.cms.waikato.ac.nz:443/open-mmlab/mmsegmentation:0.30.0_cpu
- MMSegmentation 1.1.0, PyTorch 1.9 and CUDA 11.1
  
  public.aml-repo.cms.waikato.ac.nz:443/open-mmlab/mmsegmentation:1.1.0_cuda11.1
- MMSegmentation 1.1.0, PyTorch 1.9 and CPU
  
  public.aml-repo.cms.waikato.ac.nz:443/open-mmlab/mmsegmentation:1.1.0_cpu
- DEXTR 0.1.2, PyTorch 1.10.0 and CUDA 11.1
  
  public.aml-repo.cms.waikato.ac.nz:443/pytorch/pytorch-dextr:0.1.2_cuda11.1
- DEXTR 0.1.2, PyTorch 1.9.1 and CPU
  
  public.aml-repo.cms.waikato.ac.nz:443/pytorch/pytorch-dextr:0.1.2_cpu
- Segment-Anything Model 2023-04-16, PyTorch 1.13.0 and CUDA 11.6
  
  public.aml-repo.cms.waikato.ac.nz:443/pytorch/pytorch-sam:2023-04-16_cuda11.6
- Segment-Anything Model 2023-04-16, PyTorch 1.9.1 and CPU
  
  public.aml-repo.cms.waikato.ac.nz:443/pytorch/pytorch-sam:2023-04-16_cpu
- Segment-Anything in High Quality 2023-08-17, PyTorch 1.13.0 and CUDA 11.6
  
  public.aml-repo.cms.waikato.ac.nz:443/pytorch/pytorch-sam-hq:2023-08-17_cuda11.6
- Segment-Anything in High Quality 2023-08-17, PyTorch 1.9.1 and CPU
  
  public.aml-repo.cms.waikato.ac.nz:443/pytorch/pytorch-sam-hq:2023-08-17_cpu

PyTorch/Detectron2
- Detectron2 0.3, PyTorch 1.6, CUDA 10.1
  
  public.aml-repo.cms.waikato.ac.nz:443/pytorch/detectron2:0.3
- Detectron2 0.5, PyTorch 1.9, CUDA 11.1
  
  public.aml-repo.cms.waikato.ac.nz:443/pytorch/detectron2:0.5
- Detectron2 0.6, PyTorch 1.9, CUDA 11.1
  
  public.aml-repo.cms.waikato.ac.nz:443/pytorch/detectron2:0.6
PyTorch/YOLACT++
- YOLACT++ 2020-02-11, PyTorch 1.2, CUDA 10.0
  
  public.aml-repo.cms.waikato.ac.nz:443/yolact/yolactpp:2020-02-11
PyTorch/Yolov5
- Yolov5 2022-11-05, CPU
  
  public.aml-repo.cms.waikato.ac.nz:443/pytorch/pytorch-yolov5:2022-11-05_cpu
- Yolov5 2022-11-05, CUDA 11.1
  
  public.aml-repo.cms.waikato.ac.nz:443/pytorch/pytorch-yolov5:2022-11-05_cuda11.1
Tensorflow/Object Detection API
- Tensorflow 1.14, Object Detection API 2019-08-31, CUDA 10.0
  
  public.aml-repo.cms.waikato.ac.nz:443/tensorflow/tf_object_detection:1.14.0_2019-08-31

Falcontune
- Falcontune 20230618, CUDA 11.7
  
  public.aml-repo.cms.waikato.ac.nz:443/pytorch/pytorch-huggingface-transformers:4.31.0_cuda11.7_falcontune_20230618
finetune-gpt2xl
- finetune-gpt2xl 20220924, CUDA 11.1
  
  public.aml-repo.cms.waikato.ac.nz:443/pytorch/pytorch-huggingface-transformers:4.7.0_cuda11.1_finetune-gpt2xl_20220924
Llama-2
- Falcontune 20230618, CUDA 11.7
  
  public.aml-repo.cms.waikato.ac.nz:443/pytorch/pytorch-huggingface-transformers:4.31.0_cuda11.7_llama2
XTuner
- XTuner 2024-02-19, CUDA 11.7
  
  public.aml-repo.cms.waikato.ac.nz:443/pytorch/pytorch-xtuner:2024-02-19_cuda11.7
- XTuner 0.1.15, CUDA 11.7
  
  public.aml-repo.cms.waikato.ac.nz:443/pytorch/pytorch-xtuner:0.1.15_cuda11.7
- XTuner 0.1.18, CUDA 11.7
  
  public.aml-repo.cms.waikato.ac.nz:443/pytorch/pytorch-xtuner:0.1.18_cuda11.7

Coqui STT
- Coqui STT 1.3.0, CUDA 11.0
  
  public.aml-repo.cms.waikato.ac.nz:443/tensorflow/tf_coqui_stt:1.3.0_cuda11.0
- Coqui STT 1.3.0, CPU
  
  public.aml-repo.cms.waikato.ac.nz:443/tensorflow/tf_coqui_stt:1.3.0_cpu

Huggingface
- Huggingface 4.36.0, CUDA 11.7
  
  public.aml-repo.cms.waikato.ac.nz:443/pytorch/pytorch-huggingface-transformers:4.36.0_cuda11.7_classification