chore: upgrade to cu121 because fast-whisper need this

ref: SYSTRAN/faster-whisper#717 (comment) - Update PyTorch extra-index-url from cu118 to cu121 in `Dockerfile` - Update PyTorch extra-index-url from cu118 to cu121 in `ubi.Dockerfile` Signed-off-by: 陳鈞 <[email protected]>
jim60105 · Aug 25, 2024 · a78d3dd · a78d3dd
1 parent c4aed20
commit a78d3dd
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -58,7 +58,7 @@ ARG PIP_DISABLE_PIP_VERSION_CHECK="true"
 # Install requirements
 RUN --mount=type=cache,id=pip-$TARGETARCH$TARGETVARIANT,sharing=locked,target=/root/.cache/pip \
     pip install -U --force-reinstall pip setuptools wheel && \
-    pip install -U --extra-index-url https://download.pytorch.org/whl/cu118 \
+    pip install -U --extra-index-url https://download.pytorch.org/whl/cu121 \
     torch==2.2.2 torchaudio==2.2.2 \
     pyannote.audio==3.1.1 \
     # https://github.com/jim60105/docker-whisperX/issues/40

diff --git a/ubi.Dockerfile b/ubi.Dockerfile
@@ -65,7 +65,7 @@ ARG PIP_DISABLE_PIP_VERSION_CHECK="true"
 # Install requirements
 RUN --mount=type=cache,id=pip-$TARGETARCH$TARGETVARIANT,sharing=locked,target=/root/.cache/pip \
     pip3.11 install -U --force-reinstall pip setuptools wheel && \
-    pip3.11 install -U --extra-index-url https://download.pytorch.org/whl/cu118 \
+    pip3.11 install -U --extra-index-url https://download.pytorch.org/whl/cu121 \
     torch==2.2.2 torchaudio==2.2.2 \
     pyannote.audio==3.1.1 \
     # https://github.com/jim60105/docker-whisperX/issues/40