Update travis to build against stable PyTorch
@@ -8,21 +8,19 @@ python:
- "3.6"
env:
- PYTORCH_VERSION=master
- - PYTORCH_VERSION=dummy # Just a hack to make sure th matrix runs
+ - PYTORCH_VERSION=stable
install:
- - if [[ $PYTORCH_VERSION != "dummy" ]]; then
+ - if [[ $PYTORCH_VERSION = "master" ]]; then
pip install torch_nightly -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html;
python setup.py build develop;
+ else
+ pip3 install https://download.pytorch.org/whl/cpu/torch-1.0.0-cp36-cp36m-linux_x86_64.whl;
fi
script:
- - if [[ $PYTORCH_VERSION != "dummy" ]]; then
- python -c "import torch; print('PyTorch Version:', torch.__version__)";
- python -m unittest discover;
- else
- echo "Dummy build";
- fi
+ - python -c "import torch; print('PyTorch Version:', torch.__version__)"
+ - python -m unittest discover
matrix:
include:
