upgrade pytorch in travis

This commit is contained in:
Russi Chatterjee 2019-04-05 13:45:02 +05:30
parent 9b3f68fbfd
commit db74f8ea57

View File

@ -8,7 +8,7 @@ before_install:
- sudo apt-get install -yqq libopenblas-dev liblapack3 python3-numpy python3-dev swig
- sudo ln -s /usr/lib/libopenblas.so /usr/lib/libopenblas.so.3
install:
- pip install -qqq http://download.pytorch.org/whl/cu75/torch-0.2.0.post3-cp36-cp36m-manylinux1_x86_64.whl
- pip install -qqq https://download.pytorch.org/whl/cu100/torch-1.0.1.post2-cp36-cp36m-linux_x86_64.whl
- pip install -qqq numpy
- pip install -qqq visdom
- pip install -qqq pyflann3