Skip to content

GitLab

"torchvision/git@developer.sourcefind.cn:OpenDAS/vision.git" did not exist on "b3cb86f295c8b8af7142e9def3352c5e16e481b6"
Commit db7ebce3 authored by rusty1s's avatar rusty1s
Browse files

matrix

parent 3a544b5a
Hide whitespace changes
Inline Side-by-side
Showing with 20 additions and 43 deletions
.travis.yml
View file @ db7ebce3
language: python language: python
python:
- "3.6"
- "3.7"
- "3.8"
sudo: enabled sudo: enabled
compiler: compiler:
- gcc - gcc
matrix: jobs:
include: include:
- name: cu92 - os: linux
env: python:
- FORCE_CUDA=1 - '3.6'
- CUDA=9.2.148-1 - '3.7'
- CUDA_SHORT=9.2
- UBUNTU_VERSION=ubuntu1604
- CUBLAS=cuda-cublas-dev-${CUDA_SHORT/./-}
- TORCH=1.4.0+cu92
- name: cu100
env: env:
- FORCE_CUDA=1 - FORCE_CUDA=1 CUDA=9.2.148-1 CUDA_SHORT=9.2 UBUNTU_VERSION=ubuntu1604 CUBLAS=cuda-cublas-dev-9-2 TORCH=1.4.0+cu92
- CUDA=10.0.130-1 - FORCE_CUDA=1 CUDA=10.0.130-1 CUDA_SHORT=10.0 UBUNTU_VERSION=ubuntu1804 CUBLAS=cuda-cublas-dev-10-0 TORCH=1.4.0+cu100
- CUDA_SHORT=10.0 - FORCE_CUDA=1 CUDA=10.1.105-1 CUDA_SHORT=10.1 UBUNTU_VERSION=ubuntu1804 CUBLAS=libcublas-dev TORCH=1.4.0
- UBUNTU_VERSION=ubuntu1804 before_install:
- CUBLAS=cuda-cublas-dev-${CUDA_SHORT/./-} - INSTALLER=cuda-repo-${UBUNTU_VERSION}_${CUDA}_amd64.deb
- TORCH=1.4.0+cu100 - wget http://developer.download.nvidia.com/compute/cuda/repos/${UBUNTU_VERSION}/x86_64/${INSTALLER}
- name: cu101 - sudo dpkg -i ${INSTALLER}
env: - wget https://developer.download.nvidia.com/compute/cuda/repos/${UBUNTU_VERSION}/x86_64/7fa2af80.pub
- FORCE_CUDA=1 - sudo apt-key add 7fa2af80.pub
- CUDA=10.1.105-1 - sudo apt update -qq
- CUDA_SHORT=10.1 - sudo apt install -y cuda-core-${CUDA_SHORT/./-} cuda-cudart-dev-${CUDA_SHORT/./-} ${CUBLAS} cuda-cusparse-dev-${CUDA_SHORT/./-}
- UBUNTU_VERSION=ubuntu1804 - sudo apt clean
- CUBLAS=libcublas-dev - CUDA_HOME=/usr/local/cuda-${CUDA_SHORT}
- TORCH=1.4.0 - LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
- PATH=${CUDA_HOME}/bin:${PATH}
before_install:
- INSTALLER=cuda-repo-${UBUNTU_VERSION}_${CUDA}_amd64.deb
- wget http://developer.download.nvidia.com/compute/cuda/repos/${UBUNTU_VERSION}/x86_64/${INSTALLER}
- sudo dpkg -i ${INSTALLER}
- wget https://developer.download.nvidia.com/compute/cuda/repos/${UBUNTU_VERSION}/x86_64/7fa2af80.pub
- sudo apt-key add 7fa2af80.pub
- sudo apt update -qq
- sudo apt install -y cuda-core-${CUDA_SHORT/./-} cuda-cudart-dev-${CUDA_SHORT/./-} ${CUBLAS} cuda-cusparse-dev-${CUDA_SHORT/./-}
- sudo apt clean
- CUDA_HOME=/usr/local/cuda-${CUDA_SHORT}
- LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
- PATH=${CUDA_HOME}/bin:${PATH}
install: install:
- pip install numpy - pip install numpy
...@@ -56,7 +34,6 @@ install: ...@@ -56,7 +34,6 @@ install:
- pip install flake8 - pip install flake8
script: script:
- nvcc --version
- python -c "import torch; print(torch.__version__)" - python -c "import torch; print(torch.__version__)"
- flake8 . - flake8 .
- python setup.py install - python setup.py install
......
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