do not install torchaudio and torchvision

2023-10-27 16:17:50 +02:00
parent feb34b8709
commit 9962d02872
1 changed files with 1 additions and 1 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 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
+    && pip3 install torch --index-url https://download.pytorch.org/whl/cu118

 WORKDIR /app