Commit 82ba05b5 authored by rusty1s's avatar rusty1s
Browse files

wheelhouse

parent 0381182c
......@@ -7,6 +7,7 @@ python:
os:
- linux
- osx
env:
global:
......@@ -14,21 +15,29 @@ env:
jobs:
- FORCE_CUDA=0 TORCH=1.4.0+cpu
- FORCE_CUDA=1 CUDA=9.2.148-1 CUDA_SHORT=9.2 UBUNTU_VERSION=ubuntu1604 CUBLAS=cuda-cublas-dev-9-2 TORCH=${TORCH_VERSION}+cu92
- FORCE_CUDA=1 CUDA=10.0.130-1 CUDA_SHORT=10.0 UBUNTU_VERSION=ubuntu1804 CUBLAS=cuda-cublas-dev-10-0 TORCH=${TORCH_VERSION}+cu100
- FORCE_CUDA=1 CUDA=10.1.105-1 CUDA_SHORT=10.1 UBUNTU_VERSION=ubuntu1804 CUBLAS=libcublas-dev TORCH=${TORCH_VERSION}
# - FORCE_CUDA=1 CUDA=10.0.130-1 CUDA_SHORT=10.0 UBUNTU_VERSION=ubuntu1804 CUBLAS=cuda-cublas-dev-10-0 TORCH=${TORCH_VERSION}+cu100
# - FORCE_CUDA=1 CUDA=10.1.105-1 CUDA_SHORT=10.1 UBUNTU_VERSION=ubuntu1804 CUBLAS=libcublas-dev TORCH=${TORCH_VERSION}
jobs:
exclude:
- os: osx
env: FORCE_CUDA=1
before_install:
- source install.sh
install:
- pip install numpy
- pip install torch==${TORCH} -f https://download.pytorch.org/whl/torch_stable.html
- python -c "import torch; print(torch.__version__)"
- pip install flake8
- pip install twine cibuildwheel
install:
- python setup.py install
script:
- flake8 .
- python setup.py install
- python setup.py test
- python -m cbuildwheel --output-dir wheelhouse
- ls wheelhouse
notifications:
email: false
#!/bin/bash
echo "${TRAVIS_OS_NAME}"
echo "${FORCE_CUDA}"
if [ "${TRAVIS_OS_NAME}" = "linux" ] && [ "${FORCE_CUDA}" == "1" ]; then
INSTALLER=cuda-repo-${UBUNTU_VERSION}_${CUDA}_amd64.deb
wget "http://developer.download.nvidia.com/compute/cuda/repos/${UBUNTU_VERSION}/x86_64/${INSTALLER}"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment