From e7dce17ce85ce522bf65f10ff118e2667ea0e673 Mon Sep 17 00:00:00 2001 From: Aristeu Junior Date: Sat, 16 Mar 2024 11:54:53 -0300 Subject: [PATCH] Update Dockerfile.transcriptor - cuda change --- Dockerfile.transcriptor | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/Dockerfile.transcriptor b/Dockerfile.transcriptor index 3c129bc..ccfbeef 100644 --- 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 "#####################################" \