2.3.1-cuda11.8-cudnn8-devel

Signed-off-by: siaimes <34199488+siaimes@users.noreply.github.com>

.github/workflows/build-and-push-docker-image.yml

@@ -15,23 +15,41 @@ jobs:
     runs-on: ubuntu-latest
     name: Push Docker image to Docker Hub
     steps:
+      - name: Clean runner
+        run: |
+          docker builder prune -f
+          docker system prune -a -f
+          rm -rf /usr/share/dotnet
+          rm -rf /opt/ghc
+          rm -rf "/usr/local/share/boost"
+          rm -rf "$AGENT_TOOLSDIRECTORY"
+
       - name: Check out the repo
         uses: actions/checkout@v3
 
+      - name: Add hosts to /etc/hosts
+        env:
+          super_secret: ${{ secrets.REGISTRY_HOSTIP}}
+        if: ${{ env.super_secret != '' }}
+        run: |
+          echo "${{ secrets.REGISTRY_HOSTIP}} ${{ secrets.REGISTRY_HOSTNAME}}" | sudo tee -a /etc/hosts
+          cat /etc/hosts
+
       - name: Log in to Docker Hub
         if: github.event_name == 'release'
         uses: docker/login-action@v2
         with:
-          username: ${{ secrets.DOCKERHUB_USERNAME }}
-          password: ${{ secrets.DOCKERHUB_PASSWORD }}
+          registry: ${{ secrets.REGISTRY_HOSTNAME || 'docker.io' }}
+          username: ${{ secrets.REGISTRY_USERNAME }}
+          password: ${{ secrets.REGISTRY_PASSWORD }}
 
       - name: Docker meta
         id: meta
         uses: docker/metadata-action@v4
         with:
           # list of Docker images to use as base name for tags
           images: |
-            ${{ github.event_name == 'pull_request' && github.event.pull_request.user.login || secrets.DOCKERHUB_USERNAME }}/${{ secrets.DOCKERHUB_REPONAME }}
+            ${{ secrets.REGISTRY_HOSTNAME || 'docker.io' }}/${{ github.event_name == 'pull_request' && github.event.pull_request.user.login || secrets.REGISTRY_USERNAME }}/${{ secrets.REGISTRY_REPONAME || github.event.repository.name }}
           flavor: |
             latest=auto
           # generate Docker tags based on the following events/attributes
@@ -42,6 +60,7 @@ jobs:
       - name: Push to Docker Hub
         uses: docker/build-push-action@v4
         with:
+          context: .
           push: ${{ github.event_name == 'release' }}
           tags: ${{ steps.meta.outputs.tags }}
           labels: ${{ steps.meta.outputs.labels }}

Dockerfile

@@ -1,8 +1,8 @@
-FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-devel
+FROM pytorch/pytorch:2.3.1-cuda11.8-cudnn8-devel
 
 RUN apt-get update && apt-get install -y libgl1-mesa-glx libpci-dev curl nano psmisc zip git && apt-get --fix-broken install -y
 
-RUN conda install -y faiss-gpu scikit-learn pandas flake8 yapf isort yacs gdown future libgcc -c conda-forge
+RUN conda install -y scikit-learn pandas flake8 yapf isort yacs future libgcc
 
 RUN pip install --upgrade pip && python -m pip install --upgrade setuptools && \
     pip install opencv-python tb-nightly matplotlib logger_tt tabulate tqdm wheel mccabe scipy

README.md

@@ -6,7 +6,7 @@
 
 ## Customization
 
-1. Fork This project to your own GitHub account.
-2. Add the following three secrets: `DOCKERHUB_USERNAME`, `DOCKERHUB_PASSWORD`, `DOCKERHUB_REPONAME`.
-3. Add the packages you need to `apt-get install`, `conda install` or `pip install` commands in the branch that corresponds to the PyTorch version. Afterwards, navigate to Actions tab to review the log;
-4. If there are no errors when executing the actions, proceed to release a version based on this branch. Then, visit the Actions tab to access the log and DockerHub to examine the associated tags.
+1. Fork this project to your own GitHub account.
+2. Add the following secrets: `REGISTRY_REPONAME`, `REGISTRY_USERNAME`, `REGISTRY_PASSWORD`.
+3. Add the packages you need to `apt-get install`, `conda install` or `pip install` commands in the branch that corresponds to the PyTorch version. Afterwards, navigate to `Actions` tab to review the log;
+4. If there are no errors when executing the actions, proceed to release a version based on this branch. Then, visit the `Actions` tab to access the log and DockerHub to examine the associated tags.