From 9f49275f1bc85423adcd9ee2c7d14bbff588a2bd Mon Sep 17 00:00:00 2001 From: jiltseb Date: Wed, 23 Oct 2024 20:37:12 +0000 Subject: [PATCH 1/2] removed cudnn8 dependency --- requirements.txt | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/requirements.txt b/requirements.txt index a684d4be..b6161add 100644 --- a/requirements.txt +++ b/requirements.txt @@ -3,6 +3,6 @@ huggingface_hub>=0.13 tokenizers>=0.13,<1 onnxruntime>=1.14,<2 pyannote-audio>=3.1.1 -torch>=2.1.1,<2.4.0 -torchaudio>=2.1.2,<2.4.0 +torch>=2.1.1 +torchaudio>=2.1.2 tqdm \ No newline at end of file From e72a6ea8528b34fe53850195f7848f62d266e29a Mon Sep 17 00:00:00 2001 From: jiltseb Date: Wed, 23 Oct 2024 20:44:48 +0000 Subject: [PATCH 2/2] Update dockerfile to be compatible with CT2 4.5.0 --- docker/Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docker/Dockerfile b/docker/Dockerfile index 604c8e17..91ffbf6d 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -1,4 +1,4 @@ -FROM nvidia/cuda:12.2.2-cudnn8-runtime-ubuntu22.04 +FROM nvidia/cuda:12.3.2-cudnn9-runtime-ubuntu22.04 WORKDIR /root RUN apt-get update -y && apt-get install -y python3-pip COPY infer.py jfk.flac ./