Try to fix faulty poetry / Pytorch combo.

MatthiasZepper · Nov 9, 2023 · 7504bbf · 7504bbf
1 parent 67fe6e4
commit 7504bbf
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 1,183 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -14,7 +14,7 @@ RUN pip3 install torch torchaudio --extra-index-url https://download.pytorch.org
 WORKDIR "/whisper_asr"
 COPY . "/whisper_asr"
 
-RUN poetry config virtualenvs.create false && poetry lock
+RUN poetry config virtualenvs.create false && poetry lock --no-update
 RUN poetry install --only main --no-root
 
 # cache the model of choice to allow for offline use.

diff --git a/Dockerfile.gpu b/Dockerfile.gpu
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:11.8.0-runtime-ubuntu22.04
+FROM nvidia/cuda:12.2.2-runtime-ubuntu22.04
 
 ARG PYTHON_VERSION=3.10
 ARG WHISPER_MODEL
@@ -18,7 +18,7 @@ RUN ln -s -f /usr/bin/python${PYTHON_VERSION} /usr/bin/python && \
 WORKDIR /whisper_asr
 COPY . /whisper_asr
 
-RUN poetry config virtualenvs.create false && poetry lock
+RUN poetry config virtualenvs.create false && poetry lock --no-update
 RUN poetry install --only main --no-root
 
 # cache the model of choice to allow for offline use.