Updated gpu dockerfile and docker tags

nomadkaraoke · Dec 22, 2023 · 04e6591 · 04e6591
1 parent 481ac79
commit 04e6591
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/.github/workflows/publish.yml b/.github/workflows/publish.yml
@@ -59,7 +59,7 @@ jobs:
           push: true
           tags: |
             beveradb/audio-separator:cpu-${{ env.VERSION }}
-            beveradb/audio-separator:cpu-latest
+            beveradb/audio-separator:cpu
             beveradb/audio-separator:latest
 
       - name: Build and push Docker image for GPU
@@ -72,4 +72,4 @@ jobs:
           push: true
           tags: |
             beveradb/audio-separator:gpu-${{ env.VERSION }}
-            beveradb/audio-separator:gpu-latest
+            beveradb/audio-separator:gpu
diff --git a/Dockerfile.gpu b/Dockerfile.gpu
@@ -6,14 +6,15 @@ WORKDIR /usr/src/app
 
 RUN apt-get update && apt-get install -y \
     ffmpeg \
+    python3 \
     python3-pip \
     && rm -rf /var/lib/apt/lists/*
 
-RUN python -m pip install --upgrade pip
+RUN python3 -m pip install --upgrade pip
 
 # Install the GPU version of audio-separator
 RUN --mount=type=cache,target=/root/.cache \
-    pip install "audio-separator[gpu]"
+    pip3 install "audio-separator[gpu]"
 
 # Run audio-separator when the container launches
 ENTRYPOINT ["audio-separator"]