From adae53957e12601a8d0bc4746d4c891005760352 Mon Sep 17 00:00:00 2001
From: Theodore Papamarkou <theodore.papamarkou@gmail.com>
Date: Tue, 1 Dec 2020 16:35:09 +0200
Subject: [PATCH] Further edit in meta.yml

---
 installation/meta.yaml | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

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: