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/

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
PyTorch/Yolov10
- Yolov10 2024-06-23, CPU
  
  public.aml-repo.cms.waikato.ac.nz:443/pytorch-yolov10:2024-06-23_cpu
- Yolov10 2024-06-23, CUDA 11.7
  
  waikatodatpublic.aml-repo.cms.waikato.ac.nz:443amining/pytorch-yolov10:2024-06-23_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

OpenAI Whisper
- Faster Whisper 1.0.2, CUDA 12.1
  
  public.aml-repo.cms.waikato.ac.nz:443/pytorch/python-faster-whisper:1.0.2_cuda12.1
- Faster Whisper 1.0.2, CPU
  
  public.aml-repo.cms.waikato.ac.nz:443/pytorch/python-faster-whisper:1.0.2_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

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.