diff --git a/pages/docker-images-dockerhub.rst b/pages/docker-images-dockerhub.rst index f80052d..5dc36d8 100644 --- a/pages/docker-images-dockerhub.rst +++ b/pages/docker-images-dockerhub.rst @@ -67,6 +67,19 @@ preparation, training and making predictions. +
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
Tensorflow (using our wai.tfimageclass Python library)
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)