improve torch installation and bum buildx

pluja · Oct 27, 2023 · 16d4541 · 16d4541
1 parent de54570
commit 16d4541
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.github/workflows/docker-transcription-api-gpu.yml b/.github/workflows/docker-transcription-api-gpu.yml
@@ -17,7 +17,7 @@ jobs:
         uses: docker/setup-qemu-action@v3
 
       - name: Set up Docker Buildx
-        uses: docker/setup-buildx-action@v2
+        uses: docker/setup-buildx-action@v3
 
       - name: Login to DockerHub
         uses: docker/login-action@v3

diff --git a/transcription-api/Dockerfile.gpu b/transcription-api/Dockerfile.gpu
@@ -19,7 +19,7 @@ RUN ln -s -f /usr/bin/python${PYTHON_VERSION} /usr/bin/python3 && \
 COPY requirements.txt requirements.txt
 RUN pip3 install -r requirements.txt \
     && pip3 install python-multipart \
-    && pip3 install torch==1.13.0+cu117 -f https://download.pytorch.org/whl/torch
+    && pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
 
 WORKDIR /app