diff --git a/pyproject.toml b/pyproject.toml
index 10b65275165..732a00b6047 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -61,6 +61,8 @@ dependencies=[
   "sympy",
   "tabulate",
   "torch==2.3",
+  "torchvision==0.18",
+  "torchaudio==2.3",
 ]
 
 [project.urls]