From 16d4541a47f50b10e34bf0b3a8cefe9cbd8bf437 Mon Sep 17 00:00:00 2001 From: pluja Date: Fri, 27 Oct 2023 07:53:06 +0200 Subject: [PATCH] improve torch installation and bum buildx --- .github/workflows/docker-transcription-api-gpu.yml | 2 +- transcription-api/Dockerfile.gpu | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/docker-transcription-api-gpu.yml b/.github/workflows/docker-transcription-api-gpu.yml index 9564874..6c07742 100644 --- 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 index 7e86a5d..154eb78 100644 --- 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