diff --git a/.github/workflows/docker-publish.yml b/.github/workflows/docker-publish.yml new file mode 100644 index 0000000..0bd10ea --- /dev/null +++ b/.github/workflows/docker-publish.yml @@ -0,0 +1,32 @@ +name: Docker Build and Publish + +on: + push: + branches: + - main + workflow_dispatch: + +jobs: + build-and-push-image: + runs-on: ubuntu-latest + steps: + - name: Checkout + uses: actions/checkout@v2 + + - name: Build amd64 image + run: docker buildx build --progress=plain --platform linux/amd64 -t charliecaptain/meowai-image:latest -f ./Dockerfile . + + - name: Build arm64 image + run: docker buildx build --progress=plain --platform linux/arm64 -t charliecaptain/meowai-image:latest-arm-linux -f ./Dockerfile . + + - name: Login to Docker Hub + uses: docker/login-action@v1 + with: + username: ${{ secrets.DOCKER_USERNAME }} + password: ${{ secrets.DOCKER_PASSWORD }} + + - name: Push amd64 image + run: docker push charliecaptain/meowai-image:latest + + - name: Push arm64 image + run: docker push charliecaptain/meowai-image:latest-arm-linux diff --git a/Dockerfile b/Dockerfile index b7af664..65135ad 100644 --- a/Dockerfile +++ b/Dockerfile @@ -8,17 +8,6 @@ COPY . /app WORKDIR /app # 安装所需依赖 -#RUN apt-get update --fix-missing \ -# && apt-get install ffmpeg libsm6 libxext6 -y \ -# && pip install --upgrade pip \ -# && pip install -r requirements.txt \ -# && pip install -r yolov5/requirements.txt \ -# && pip install torch==1.13.0 torchvision==0.14.0 --index-url https://download.pytorch.org/whl/cpu \ -## && apt-get install -y libgl1-mesa-glx \ -# # 删除缓存文件 -# && rm -rf /var/lib/apt/lists/* \ -# && rm -rf ~/.cache/pip - RUN pip install --upgrade pip \ && pip install -r requirements.txt \ && pip install torch==1.13.0 torchvision==0.14.0 --index-url https://download.pytorch.org/whl/cpu \