on TravisCI, test with pytorch-cpu is enough

lanpa · Jan 31, 2018 · ad7d652 · ad7d652
1 parent 0c9f7ba
commit ad7d652
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.travis.yml b/.travis.yml
@@ -22,7 +22,7 @@ install:
   - conda info -a
 
   # Replace dep1 dep2 ... with your dependencies
-  - conda create -q -n test-environment python=$TRAVIS_PYTHON_VERSION chainer pytorch torchvision -c pytorch
+  - conda create -q -n test-environment python=$TRAVIS_PYTHON_VERSION chainer pytorch-cpu torchvision -c pytorch
   - source activate test-environment
   - which python
   - conda list