forked from stanfordnlp/spinn
-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy path.travis.yml
27 lines (23 loc) · 1022 Bytes
/
.travis.yml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
language: python
dist: trusty
python: "2.7"
sudo: true
env: CUDA=7.5-18 THEANO_FLAGS=device=gpu,floatX=float32
before_install:
# Install CUDA toolkit
- echo "Installing CUDA library"
- travis_retry wget http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1404/x86_64/cuda-repo-ubuntu1404_${CUDA}_amd64.deb
- travis_retry sudo dpkg -i cuda-repo-ubuntu1404_${CUDA}_amd64.deb
- travis_retry sudo apt-get update -qq
- export CUDA_APT=${CUDA%-*}
- export CUDA_APT=${CUDA_APT/./-}
# - travis_retry sudo apt-get install -y cuda-${CUDA_APT}
- travis_retry sudo apt-get install -y nvidia-settings cuda-drivers cuda-core-${CUDA_APT} cuda-cudart-dev-${CUDA_APT}
- travis_retry sudo apt-get clean
- export CUDA_HOME=/usr/local/cuda-${CUDA%%-*}
- export LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
- export PATH=${CUDA_HOME}/bin:${PATH}
install:
- nvcc --version
- pip install -r python/requirements.txt
script: "nosetests -a \\!slow python/spinn/tests"