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 ./
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