diff --git a/tests/torch2/requirements.txt b/tests/torch2/requirements.txt
index a7ad0bba674..5ac1e0ffd0f 100644
--- a/tests/torch2/requirements.txt
+++ b/tests/torch2/requirements.txt
@@ -9,3 +9,5 @@ pytest-mock
 pytest-xdist
 torch
 torchvision
+
+efficientnet_pytorch==0.7.1