Skip to content

Commit

Permalink
added mmpretrain docker images and post announcing them
Browse files Browse the repository at this point in the history
  • Loading branch information
fracpete committed Mar 13, 2024
1 parent d54cd77 commit 0a606f0
Show file tree
Hide file tree
Showing 3 changed files with 43 additions and 0 deletions.
13 changes: 13 additions & 0 deletions pages/docker-images-dockerhub.rst
Original file line number Diff line number Diff line change
Expand Up @@ -67,6 +67,19 @@ preparation, training and making predictions.
</li>
</ul>
</li>
<li>
<p>PyTorch/MMPretrain (successor of MMClassification)</p>
<ul>
<li>
<p><a class="reference external" href="https://github.com/waikato-datamining/mmpretrain/blob/master/1.2.0_cuda11.1">MMPretrain 1.2.0, PyTorch 1.9 and CUDA 11.1</a></p>
<p><code class="docutils literal"><span class="pre">waikatodatamining/mmpretrain:1.2.0_cuda11.1</span></code></p>
</li>
<li>
<p><a class="reference external" href="https://github.com/waikato-datamining/mmpretrain/blob/master/1.2.0_cpu">MMPretrain 1.2.0, PyTorch 1.11 and CPU</a></p>
<p><code class="docutils literal"><span class="pre">waikatodatamining/mmpretrain:1.2.0_cpu</span></code></p>
</li>
</ul>
</li>
<li>
<p>Tensorflow (using our <a class="reference external" href="https://pypi.org/project/wai.tfimageclass/">wai.tfimageclass</a> Python library)</p>
<ul>
Expand Down
13 changes: 13 additions & 0 deletions pages/docker-images-inhouse.rst
Original file line number Diff line number Diff line change
Expand Up @@ -74,6 +74,19 @@ preparation, training and making predictions.
</li>
</ul>
</li>
<li>
<p>PyTorch/MMPretrain (successor of MMClassification)</p>
<ul>
<li>
<p><a class="reference external" href="https://github.com/waikato-datamining/mmpretrain/blob/master/1.2.0_cuda11.1">MMPretrain 1.2.0, PyTorch 1.9 and CUDA 11.1</a></p>
<p><code class="docutils literal"><span class="pre">public.aml-repo.cms.waikato.ac.nz:443/open-mmlab/mmpretrain:1.2.0_cuda11.1</span></code></p>
</li>
<li>
<p><a class="reference external" href="https://github.com/waikato-datamining/mmpretrain/blob/master/1.2.0_cpu">MMPretrain 1.2.0, PyTorch 1.11 and CPU</a></p>
<p><code class="docutils literal"><span class="pre">public.aml-repo.cms.waikato.ac.nz:443/open-mmlab/mmpretrain:1.2.0_cpu</span></code></p>
</li>
</ul>
</li>
<li>
<p>Tensorflow (using our <a class="reference external" href="https://pypi.org/project/wai.tfimageclass/">wai.tfimageclass</a> Python library)</p>
<ul>
Expand Down
17 changes: 17 additions & 0 deletions posts/2024-03-14-mmpretrain_docker_120.rst
Original file line number Diff line number Diff line change
@@ -0,0 +1,17 @@
.. title: MMPretrain 1.2.0 Docker images available
.. slug: 2024-03-14-mmpretrain-docker
.. date: 2024-03-14 11:55:00 UTC+13:00
.. tags: release
.. category: docker
.. link:
.. description:
.. type: text
First Docker images are available for the `MMPretrain <https://github.com/open-mmlab/mmpretrain>`__
framework, using the 1.2.0 release of MMPretrain (code base as of 2024-01-05):

* `CUDA 11.1 <https://github.com/waikato-datamining/mmpretrain/tree/master/1.2.0_cuda11.1>`__
* `CPU <https://github.com/waikato-datamining/mmpretrain/tree/master/1.2.0_cpu>`__

**NB:** MMPretrain is the successor of MMClassification, which can be used for image classification.

0 comments on commit 0a606f0

Please sign in to comment.