Update Dockerfile.transcriptor - cuda change

iped-docker · Mar 16, 2024 · e7dce17 · e7dce17
1 parent ce17aee
commit e7dce17
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/Dockerfile.transcriptor b/Dockerfile.transcriptor
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:11.6.2-cudnn8-runtime-ubuntu20.04
+FROM nvidia/cuda:12.1.1-runtime-ubuntu20.04
 
 ARG SNAPSHOT=true IPED_RELEASE_VERSION=4.2 SNAPSHOT_WORKFLOW_ID=8243028270 PKGTMPDIR=/tmp/pkgs 
 
@@ -22,9 +22,9 @@ RUN mkdir -p ${PKGTMPDIR} && cd ${PKGTMPDIR} \
     && update-alternatives --install /usr/bin/python python /usr/bin/python3.9 1 \
     && update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.9 1 \
     && python -m pip install pip --upgrade \
-    && python -m pip install torch==1.12.1+cu116 torchvision==0.13.1+cu116 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu116 \
+    && python -m pip install torch torchvision torchaudio \
     && python -m pip install huggingsound \
-    && python -m pip install faster-whisper==0.10.1 \
+    && python -m pip install faster-whisper \
     && echo "#####################################" \
     && echo "Cleaning UP the container " \
     && echo "#####################################" \