diff --git a/pyproject.toml b/pyproject.toml
index af33458a..8ac2e386 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -63,6 +63,7 @@ seaborn = "^0.13"
 pyyaml = "^6"
 tensorboard = "^2"
 typing-extensions = "*"
+torchvision = ">=0.8.0"
 
 [tool.poetry.group.dev.dependencies]
 ruff = "^0.4"