diff --git a/Dockerfile b/Dockerfile
index 6c41053..01c0c00 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,5 +1,5 @@
-# Use ROCm PyTorch base image
-FROM rocm/pytorch:latest
+# Use ROCm PyTorch base image with compatible PyTorch
+FROM rocm/pytorch:2.4.0-rocm6.0-cxx11-ubuntu22.04
 
 # Set environment variables for ROCm and Python
 ENV PYTHONDONTWRITEBYTECODE=1
diff --git a/pyproject.toml b/pyproject.toml
index 324f4f7..27886f9 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -19,8 +19,9 @@ dependencies = [
     "av<16.0.0",
     "numpy>=2.1.0,<2.3.0; python_version >='3.13'",
     "pyannote-audio>=3.3.2,<4.0.0",
-    "torch~=2.8.0",
-    "torchaudio~=2.8.0",
+    "torch~=2.4.0",
+    "torchaudio~=2.4.0",
+    "torchvision~=0.19.0",
     "transformers>=4.48.0",
     "triton>=3.3.0; sys_platform == 'linux' and platform_machine == 'x86_64'", # only install triton on x86_64 Linux
     "fastapi>=0.104.0",