pytorch-dnc/.travis.yml

22 lines
868 B
YAML
Raw Normal View History

2017-10-30 00:32:31 +08:00
language: python
python:
- "3.6"
# command to install dependencies
2017-12-01 00:50:14 +08:00
before_install:
- sudo apt-get -qq update
2017-12-01 03:07:51 +08:00
- sudo apt-get install software-properties-common
2017-12-01 00:50:14 +08:00
- sudo apt-get install libopenblas-dev
2017-12-01 01:21:47 +08:00
- sudo ln -s /usr/lib/libopenblas.so /usr/lib/libopenblas.so.3
2017-12-01 14:00:51 +08:00
- wget https://developer.nvidia.com/compute/cuda/8.0/Prod2/local_installers/cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64-deb -O /tmp/cuda.deb
- sudo dpkg -i /tmp/cuda.deb
2017-12-01 03:30:53 +08:00
- sudo apt-get update
- sudo apt-get install cuda
2017-10-30 00:32:31 +08:00
install:
- pip install http://download.pytorch.org/whl/cu75/torch-0.2.0.post3-cp36-cp36m-manylinux1_x86_64.whl
- pip install numpy
- pip install visdom
# command to run tests
script:
2017-12-01 16:22:00 +08:00
- sudo cp -pr /home/travis/virtualenv/python3.6/lib/python3.6/site-packages/torch/lib/libgomp-ae56ecdc.so.1.0.0 /usr/lib/libgomp.so.1
2017-12-01 00:42:54 +08:00
- pip install -e .
2017-11-30 03:14:26 +08:00
- pytest ./test