diff --git a/hpvm/env.yaml b/hpvm/env.yaml
index d2dcaa3de27105ac07efd81cfdf9a37655c33af7..612cc956c1fe1c7ab7746a7dc18a59750a7aebcf 100644
--- a/hpvm/env.yaml
+++ b/hpvm/env.yaml
@@ -12,7 +12,7 @@ dependencies:
   - python==3.6.13
   - pip
   - pytorch==1.6.0
-  - torchvision=0.8
+  - torchvision=0.7
   - tqdm=4.59
   - scipy==1.1.0
   - h5py==2.10.0