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 "#####################################" \