diff --git a/environment-cpu.yml b/environment-cpu.yml
index 2b4db1974bd9e483bc8a2d8653ee47d3783a6e47..2e56db7f386c7d1340187b7c21892f198b07d2e7 100644
--- a/environment-cpu.yml
+++ b/environment-cpu.yml
@@ -12,7 +12,7 @@ dependencies:
   - numpy=1.16.4
   - pandas=0.25.1
   - pytest=5.0.1
-  - pytorch-cpu=1.1.0
+  - pytorch-cpu=1.2.0
   - pyyaml=5.1.2
   - scipy=1.3.1
   - torchvision-cpu=0.3.0
diff --git a/environment.yml b/environment.yml
index d03b0112fae4a10c2b7d95625db2260648dc5973..480603aca7d6e8518d254f9024c4d5dd0afeb54a 100644
--- a/environment.yml
+++ b/environment.yml
@@ -16,4 +16,4 @@ dependencies:
   - pytorch=1.2.0
   - pyyaml=5.1.2
   - scipy=1.3.1
-  - torchvision
+  - torchvision=0.3.0