Update .travis.yml (pytorch#791)

vcarpani · Feb 18, 2020 · 57df912 · 57df912
1 parent fceb3d7
commit 57df912
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/.travis.yml b/.travis.yml
@@ -24,10 +24,8 @@ before_install: &before_install
   - conda update -q conda
   # Useful for debugging any issues with conda
   - conda info -a
-  - conda create -q -n test-environment pytorch cpuonly python=$TRAVIS_PYTHON_VERSION -c $PYTORCH_CHANNEL
+  - conda create -q -n test-environment pytorch torchvision cpuonly python=$TRAVIS_PYTHON_VERSION -c $PYTORCH_CHANNEL
   - source activate test-environment
-  # Temporary fix until https://github.com/pytorch/pytorch/issues/33103
-  - if [[ "$PYTORCH_CHANNEL" == "pytorch-nightly" ]]; then pip install --upgrade git+https://github.com/pytorch/vision.git; else conda install torchvision cpuonly python=$TRAVIS_PYTHON_VERSION -c $PYTORCH_CHANNEL; fi
   # Test contrib dependencies
   - pip install tqdm scikit-learn matplotlib tensorboardX visdom polyaxon-client mlflow neptune-client tensorboard pynvml