Docker release (#94)

* Docker flow

* Dependency remove

* Update docker version

* Add docker hub library name

* Add docker hub library name for re-tagging

* GPU Dockerfile update

* Clean up docker hub release flow

Author: Hanbin Hu

.github/workflows/release.yml

@@ -59,3 +59,27 @@ jobs:
         run: |
           python setup.py sdist
           twine upload dist/*
+  docker:
+    name: DockerHub Release
+    needs: deploy
+    runs-on: ubuntu-latest
+    steps:
+      - name: Checkout code
+        uses: actions/checkout@v2
+      - name: Build CPU image
+        run: docker build . --file dockerfile.cpu --tag bluefog:cpu
+      - name: Build GPU image
+        run: docker build . --file dockerfile.gpu --tag bluefog:gpu
+      - name: Log into DockerHub
+        run: echo "${{ secrets.DOCKERHUB_PASSWORD }}" | docker login -u ${{ secrets.DOCKERHUB_USERNAME }} --password-stdin
+      - name: Push image
+        run: |
+          # Strip git ref prefix from version
+          VERSION=$(echo "${{ github.ref }}" | sed -e 's,.*/\(.*\),\1,')
+          # Strip "v" prefix from tag name
+          VERSION=$(echo $VERSION | sed -e 's/^v//')
+          echo VERSION=$VERSION
+          docker tag bluefog:cpu bluefoglib/bluefog:cpu-$VERSION
+          docker tag bluefog:gpu bluefoglib/bluefog:gpu-$VERSION
+          docker push bluefoglib/bluefog:cpu-$VERSION
+          docker push bluefoglib/bluefog:gpu-$VERSION

dockerfile.gpu

@@ -74,13 +74,12 @@ RUN pip install torch==${PYTORCH_VERSION} torchvision==${TORCHVISION_VERSION} -f
 RUN pip install "Pillow<7.0" --no-deps
 
 # Install Bluefog
-RUN pip install bluefog
-
-RUN mkdir /bluefog
-RUN cd /bluefog && \
+CMD pip install bluefog && \
+    cd /bluefog && \
     VERSION=`python -c "import bluefog; print(bluefog.__version__)"` && \
     wget https://github.com/Bluefog-Lib/bluefog/releases/download/v$VERSION/examples.tar.gz && \
-    tar -zxv -f examples.tar.gz && \
-    rm examples.tar.gz
+    tar -zx -f examples.tar.gz && \
+    rm examples.tar.gz && \
+    bash
 
 WORKDIR "bluefog"

docs/docker.rst

@@ -23,13 +23,13 @@ Downloading Docker Image From Docker Hub
 
 .. code-block:: bash
 
-    sudo docker pull bluefoglib/bluefog:gpu-0.2.2
+    sudo docker pull bluefoglib/bluefog:gpu-0.3.0
 
 2. Download docker image with only CPU support:
 
 .. code-block:: bash
 
-    sudo docker pull bluefoglib/bluefog:cpu-0.2.2
+    sudo docker pull bluefoglib/bluefog:cpu-0.3.0
 
 Building Your Own Docker Image
 ##############################