diff --git a/installation/meta.yaml b/installation/meta.yaml
index 0c0908f..02709fb 100644
--- a/installation/meta.yaml
+++ b/installation/meta.yaml
@@ -21,8 +21,9 @@ requirements:
     - kanga >=0.0.15
     - numpy >=1.19.2
     - python
-    - torch >=1.6.0
+    - pytorch
     - torchdiffeq >=0.1.1
+    - torchvision
 
 test:
   imports: