Further edit in meta.yml

papamarkou · Dec 1, 2020 · adae539 · adae539
1 parent 3a919c1
commit adae539
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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: