diff --git a/requirements-test.txt b/requirements-test.txt
index 5bb4e5c..3d75887 100644
--- a/requirements-test.txt
+++ b/requirements-test.txt
@@ -16,4 +16,6 @@ piexif==1.1.3
 pyfakefs
 pynvml==11.5.0
 tensorboard==2.12.0
+torch
+torchvision
 wandb