improve torch installation and bum buildx

2023-10-27 07:53:06 +02:00
parent de54570735
commit 16d4541a47
2 changed files with 2 additions and 2 deletions
--- 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