switch back to using pytorch

conda/meta.yaml

@@ -33,13 +33,13 @@ requirements:
     - bob.bio.base
     - bob.io.image
     - bob.ap
-    - numpy
-    - docopt
-    - matplotlib
-    - h5py
-    - tensorflow
-    # - pytorch-cpu
-    # - torchvision-cpu
+    - numpy {{ numpy }}
+    - docopt {{ docopt }}
+    - matplotlib {{ matplotlib }}
+    - h5py {{ h5py }}
+    - tensorflow {{ tensorflow }}
+    - pytorch {{ pytorch }}
+    - torchvision {{ torchvision }}
   run:
     - python
     - setuptools
@@ -48,6 +48,8 @@ requirements:
     - {{ pin_compatible('matplotlib') }}
     - {{ pin_compatible('h5py') }}
     - {{ pin_compatible('tensorflow') }}
+    - {{ pin_compatible('pytorch') }}
+    - {{ pin_compatible('torchvision') }}
 
 test:
   imports:
@@ -69,8 +71,6 @@ test:
     - coverage
     - sphinx
     - sphinx_rtd_theme
-    - pytorch-cpu
-    - torchvision-cpu
 
 about:
   home: https://www.idiap.ch/software/bob/