Merge pull request #40 from mobiusml/update_torch_req

removed cudnn8 dependency
mobiusml · Oct 23, 2024 · 0dfcee3 · 0dfcee3
2 parents ac994c2 + e72a6ea
commit 0dfcee3
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git 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 ./

diff --git 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