diff --git a/training-requirements.txt b/training-requirements.txt
index dea26c5807b5a0b8a70c2d0e476bc034cbfb4bc1..08ca60d0591cb5c128b1643054e1da88a7230a6e 100644
--- a/training-requirements.txt
+++ b/training-requirements.txt
@@ -4,4 +4,5 @@ mlflow==2.2.1
 Shapely==2.0.1
 teklia-toolbox==0.1.3
 tensorboard==2.12.1
+torchvision==0.15.1
 tqdm==4.65.0