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 22:18:39 +08:00
|
|
|
- sudo apt-get install -y software-properties-common git
|
|
|
|
|
- sudo apt-get install -y libopenblas-dev liblapack3 python3-numpy python3-dev swig
|
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
|
2017-12-01 22:18:39 +08:00
|
|
|
- sudo apt-get install -y cuda
|
2017-10-30 00:32:31 +08:00
|
|
|
install:
|
2017-12-01 22:18:39 +08:00
|
|
|
- pip install http://download.pytorch.org/whl/cu80/torch-0.2.0.post3-cp36-cp36m-manylinux1_x86_64.whl
|
2017-10-30 00:32:31 +08:00
|
|
|
- pip install numpy
|
|
|
|
|
- pip install visdom
|
|
|
|
|
# command to run tests
|
|
|
|
|
script:
|
2017-11-30 03:14:26 +08:00
|
|
|
- pytest ./test
|
