.travis.yml 1.64 KB
Newer Older
rusty1s's avatar
rusty1s committed
1
2
sudo: enabled

rusty1s's avatar
matrix  
rusty1s committed
3
env:
rusty1s's avatar
rusty1s committed
4
5
  global:
    - TORCH_VERSION=1.4.0
rusty1s's avatar
rusty1s committed
6
    - CIBW_BUILD=cp36-*
rusty1s's avatar
rusty1s committed
7
8
9
  # 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
10
11
    # - 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
12
13

jobs:
rusty1s's avatar
rusty1s committed
14
  include:
rusty1s's avatar
rusty1s committed
15
    - os: linux
rusty1s's avatar
fixes  
rusty1s committed
16
      language: python
rusty1s's avatar
rusty1s committed
17
      python: 3.6
rusty1s's avatar
rusty1s committed
18
19
20
      env:
        - TORCH=${TORCH_VERSION}+cpu
    - os: linux
rusty1s's avatar
fixes  
rusty1s committed
21
      language: python
rusty1s's avatar
rusty1s committed
22
23
24
25
26
27
28
29
30
31
      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
rusty1s's avatar
fix  
rusty1s committed
32
33
      env:
        - TORCH=${TORCH_VERSION}+cpu
rusty1s's avatar
rusty1s committed
34
35
36
37
38
      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
39

rusty1s's avatar
fix  
rusty1s committed
40
41
install:
  - pip3 install --upgrade pip
rusty1s's avatar
rusty1s committed
42
  - source install.sh
rusty1s's avatar
rusty1s committed
43
44
  - pip3 install numpy
  - pip3 install torch==${TORCH} -f https://download.pytorch.org/whl/torch_stable.html
rusty1s's avatar
rusty1s committed
45
  - python -c "import torch; print(torch.__version__)"
rusty1s's avatar
rusty1s committed
46
47
48
  - pip3 install flake8
  - pip3 install twine cibuildwheel
  - python3 setup.py install || python setup.py install
rusty1s's avatar
rusty1s committed
49

rusty1s's avatar
rusty1s committed
50
script:
rusty1s's avatar
rusty1s committed
51
  - flake8 .
rusty1s's avatar
rusty1s committed
52
53
  - pytest
  - python3 -m cibuildwheel --output-dir wheelhouse
rusty1s's avatar
rusty1s committed
54
  - ls wheelhouse
rusty1s's avatar
rusty1s committed
55
56
notifications:
  email: false