diff --git a/transcription-api/Dockerfile.gpu b/transcription-api/Dockerfile.gpu index 154eb78..416a8dd 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 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 + && pip3 install torch --index-url https://download.pytorch.org/whl/cu118 WORKDIR /app