.travis.yml 1.61 KB
Newer Older
rusty1s's avatar
rusty1s committed
1
2
3
4
language: python

sudo: enabled

rusty1s's avatar
matrix  
rusty1s committed
5
6
python:
  - 3.6
rusty1s's avatar
rusty1s committed
7

rusty1s's avatar
matrix  
rusty1s committed
8
9
os:
  - linux
rusty1s's avatar
rusty1s committed
10
  - windows
rusty1s's avatar
matrix  
rusty1s committed
11
12

env:
rusty1s's avatar
rusty1s committed
13
14
  global:
    - TORCH_VERSION=1.4.0
rusty1s's avatar
rusty1s committed
15
    - CIBW_BUILD=cp36-*
rusty1s's avatar
rusty1s committed
16
17
18
  # jobs:
  #   - FORCE_CUDA=0 TORCH=${TORCH_VERSION}+cpu
  #   - FORCE_CUDA=1 CUDA_SHORT=9.2 CUDA=9.2.148-1 UBUNTU_VERSION=ubuntu1604 CUBLAS=cuda-cublas-dev-9-2 TORCH=${TORCH_VERSION}+cu92
rusty1s's avatar
rusty1s committed
19
20
    # - FORCE_CUDA=1 CUDA_SHORT=10.0 CUDA=10.0.130-1 UBUNTU_VERSION=ubuntu1804 CUBLAS=cuda-cublas-dev-10-0 TORCH=${TORCH_VERSION}+cu100
    # - FORCE_CUDA=1 CUDA_SHORT=10.1 CUDA=10.1.105-1 UBUNTU_VERSION=ubuntu1804 CUBLAS=libcublas-dev TORCH=${TORCH_VERSION}
rusty1s's avatar
rusty1s committed
21
22

jobs:
rusty1s's avatar
rusty1s committed
23
  include:
rusty1s's avatar
rusty1s committed
24
    - os: linux
rusty1s's avatar
rusty1s committed
25
      python: 3.6
rusty1s's avatar
rusty1s committed
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
      env:
        - TORCH=${TORCH_VERSION}+cpu
    - os: linux
      python: 3.6
      env:
        - FORCE_CUDA=1
        - CUDA_SHORT=9.2
        - CUDA=9.2.148-1
        - UBUNTU_VERSION=ubuntu1604
        - CUBLAS=cuda-cublas-dev-9-2
        - TORCH=${TORCH_VERSION}+cu92
    - os: windows
      language: shell  # `language: python` is an error on Travis CI Windows
      before_install:
        - choco install python --version 3.6.7
    # - os: o
    #   python: 3.6
    #   env: FORCE_CUDA=0 TORCH=${TORCH_VERSION}
rusty1s's avatar
matrix  
rusty1s committed
44
45

before_install:
rusty1s's avatar
rusty1s committed
46
  - pip3 --upgrade pip
rusty1s's avatar
rusty1s committed
47
  - source install.sh
rusty1s's avatar
rusty1s committed
48
49
  - pip3 install numpy
  - pip3 install torch==${TORCH} -f https://download.pytorch.org/whl/torch_stable.html
rusty1s's avatar
rusty1s committed
50
  - python -c "import torch; print(torch.__version__)"
rusty1s's avatar
rusty1s committed
51
52
  - pip3 install flake8
  - pip3 install twine cibuildwheel
rusty1s's avatar
rusty1s committed
53
54

install:
rusty1s's avatar
rusty1s committed
55
  - python3 setup.py install || python setup.py install
rusty1s's avatar
rusty1s committed
56

rusty1s's avatar
rusty1s committed
57
script:
rusty1s's avatar
rusty1s committed
58
  - flake8 .
rusty1s's avatar
rusty1s committed
59
60
  - pytest
  - python3 -m cibuildwheel --output-dir wheelhouse
rusty1s's avatar
rusty1s committed
61
  - ls wheelhouse
rusty1s's avatar
rusty1s committed
62
63
notifications:
  email: false