Merge pull request #5 from matthewfeickert/fix/fix-Binder-support

Fix binder support by adding torchvision, torchtest, and opencv to environment.yml
Atcold · Feb 5, 2019 · 472dec8 · 472dec8
2 parents 3026645 + ce7a261
commit 472dec8
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -1,5 +1,5 @@
 # Mini Course in Deep Learning with PyTorch for AIMS
-[![Binder](https://mybinder.org/badge.svg)](https://mybinder.org/v2/gh/Atcold/PyTorch-Deep-Learning-Minicourse/master)
+[![Binder](https://mybinder.org/badge_logo.svg)](https://mybinder.org/v2/gh/Atcold/pytorch-Deep-Learning-Minicourse/master)
 
 The African Masters of Machine Intelligence (AMMI) is Africa's flagship program in machine intelligence led by The African Institute for Mathematical Sciences (AIMS).
 These lessons, developed during the course of several years while I've been teaching at Purdue and NYU, are here proposed for the AMMI (AIMS).

diff --git a/apt.txt b/apt.txt
@@ -0,0 +1 @@
+libgl1-mesa-glx
diff --git a/environment.yml b/environment.yml
@@ -4,6 +4,11 @@ channels:
   - pytorch
 dependencies:
   - python=3.6
+  - pip
   - pytorch=0.4.1
+  - torchvision
+  - opencv
   - matplotlib
   - jupyter
+  - pip:
+    - torchtext