fix travis build

1adrianb · Jan 18, 2020 · 34d2b37 · 34d2b37
1 parent 5ca46a3
commit 34d2b37
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.travis.yml b/.travis.yml
@@ -27,7 +27,7 @@ install:
   - conda create -q -n test-environment python=$TRAVIS_PYTHON_VERSION
   - source activate test-environment
   - travis_retry pip install --upgrade pip setuptools wheel
-  - conda install pytorch-cpu torchvision-cpu -c pytorch
+  - conda install pytorch torchvision cpuonly -c pytorch
   - python setup.py install
   - pip install --upgrade pytest