Skip to content

GitLab

Commit a372eb41 authored by rusty1s's avatar rusty1s
Browse files

different pythons

parent 6067286b
Hide whitespace changes
Inline Side-by-side
Showing with 127 additions and 50 deletions
.travis.yml
View file @ a372eb41
language: shell
os:
- linux
- osx
# - windows
env:
global:
- TORCH_VERSION=1.4.0
- CUDA_HOME=/usr/local/cuda
# 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
# - 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
jobs:
- TORCH_VERSION=1.4.0 PYTHON_VERSION=3.8.1 IDX=cpu
- TORCH_VERSION=1.4.0 PYTHON_VERSION=3.8.1 IDX=cu92
- TORCH_VERSION=1.4.0 PYTHON_VERSION=3.8.1 IDX=cu100
- TORCH_VERSION=1.4.0 PYTHON_VERSION=3.8.1 IDX=cu101
- TORCH_VERSION=1.4.0 PYTHON_VERSION=3.7.6 IDX=cpu
- TORCH_VERSION=1.4.0 PYTHON_VERSION=3.7.6 IDX=cu92
- TORCH_VERSION=1.4.0 PYTHON_VERSION=3.7.6 IDX=cu100
- TORCH_VERSION=1.4.0 PYTHON_VERSION=3.7.6 IDX=cu101
# - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.6.10 IDX=cpu
# - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.6.10 IDX=cu92
# - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.6.10 IDX=cu100
# - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.6.10 IDX=cu101
# - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.5.9 IDX=cpu
# - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.5.9 IDX=cu92
# - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.5.9 IDX=cu100
# - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.5.9 IDX=cu101
jobs:
include:
- name: linux/cpu
os: linux
env:
- IDX=cpu
- TOOLKIT=cpuonly
- name: linux/cu92
os: linux
env:
- IDX=cu92
- CUDA_SHORT=9.2
- CUDA=9.2.148-1
- UBUNTU_VERSION=ubuntu1604
- CUBLAS=cuda-cublas-dev-9-2
- TOOLKIT="cudatoolkit=${CUDA_SHORT}"
- name: linux/cu100
os: linux
env:
- IDX=cu100
- CUDA_SHORT=10.0
- CUDA=10.0.130-1
- UBUNTU_VERSION=ubuntu1804
- CUBLAS=cuda-cublas-dev-10-0
- TOOLKIT="cudatoolkit=${CUDA_SHORT}"
- name: linux/cu101
os: linux
env:
- IDX=cu101
- CUDA_SHORT=10.1
- CUDA=10.1.105-1
- UBUNTU_VERSION=ubuntu1804
- CUBLAS=libcublas-dev
- TOOLKIT="cudatoolkit=${CUDA_SHORT}"
- name: macosx/cpu
os: osx
env:
- IDX=cpu
- TOOLKIT=""
- name: windows/cpu
os: windows
env:
- IDX=cpu
- TOOLKIT=cpuonly
exclude:
- os: osx
env: TORCH_VERSION=1.4.0 PYTHON_VERSION=3.8.1 IDX=cu92
- os: osx
env: TORCH_VERSION=1.4.0 PYTHON_VERSION=3.8.1 IDX=cu100
- os: osx
env: TORCH_VERSION=1.4.0 PYTHON_VERSION=3.8.1 IDX=cu101
- os: osx
env: TORCH_VERSION=1.4.0 PYTHON_VERSION=3.7.6 IDX=cu92
- os: osx
env: TORCH_VERSION=1.4.0 PYTHON_VERSION=3.7.6 IDX=cu100
- os: osx
env: TORCH_VERSION=1.4.0 PYTHON_VERSION=3.7.6 IDX=cu101
- os: osx
env: TORCH_VERSION=1.4.0 PYTHON_VERSION=3.6.10 IDX=cu92
- os: osx
env: TORCH_VERSION=1.4.0 PYTHON_VERSION=3.6.10 IDX=cu100
- os: osx
env: TORCH_VERSION=1.4.0 PYTHON_VERSION=3.6.10 IDX=cu101
- os: osx
env: TORCH_VERSION=1.4.0 PYTHON_VERSION=3.5.9 IDX=cu92
- os: osx
env: TORCH_VERSION=1.4.0 PYTHON_VERSION=3.5.9 IDX=cu100
- os: osx
env: TORCH_VERSION=1.4.0 PYTHON_VERSION=3.5.9 IDX=cu101
# jobs:
# include:
# - name: linux/cpu
# os: linux
# env:
# - IDX=cpu
# - TOOLKIT=cpuonly
# - name: linux/cu92
# os: linux
# env:
# - IDX=cu92
# - CUDA_SHORT=9.2
# - CUDA=9.2.148-1
# - UBUNTU_VERSION=ubuntu1604
# - CUBLAS=cuda-cublas-dev-9-2
# - TOOLKIT="cudatoolkit=${CUDA_SHORT}"
# - name: linux/cu100
# os: linux
# env:
# - IDX=cu100
# - CUDA_SHORT=10.0
# - CUDA=10.0.130-1
# - UBUNTU_VERSION=ubuntu1804
# - CUBLAS=cuda-cublas-dev-10-0
# - TOOLKIT="cudatoolkit=${CUDA_SHORT}"
# - name: linux/cu101
# os: linux
# env:
# - IDX=cu101
# - CUDA_SHORT=10.1
# - CUDA=10.1.105-1
# - UBUNTU_VERSION=ubuntu1804
# - CUBLAS=libcublas-dev
# - TOOLKIT="cudatoolkit=${CUDA_SHORT}"
# - name: macosx/cpu
# os: osx
# env:
# - IDX=cpu
# - TOOLKIT=""
# - name: windows/cpu
# os: windows
# env:
# - IDX=cpu
# - TOOLKIT=cpuonly
# - name: windows/cu92
# os: windows
# language: shell
......@@ -89,9 +133,9 @@ jobs:
# - choco install python --version 3.7.6
install:
- source script/install.sh
- source script/cuda.sh
- source script/conda.sh
- conda create --yes -n test python=3.7.6
- conda create --yes -n test python=${PYTHON_VERSION}
- source activate test
- conda install pytorch=${TORCH_VERSION} ${TOOLKIT} -c pytorch --yes
- pip install flake8 codecov
......
script/install.sh → script/cuda.sh
View file @ a372eb41
#!/bin/bash
if [ "${TRAVIS_OS_NAME}" = "linux" ] && [ "$IDX" = "cpu" ]; then
export TOOLKIT=cpuonly
fi
if [ "${TRAVIS_OS_NAME}" = "linux" ] && [ "$IDX" = "cu92" ]; then
export CUDA_SHORT=9.2
export CUDA=9.2.148-1
export UBUNTU_VERSION=ubuntu1604
export CUBLAS=cuda-cublas-dev-9-2
export TOOLKIT="cudatoolkit=${CUDA_SHORT}"
fi
if [ "${TRAVIS_OS_NAME}" = "linux" ] && [ "$IDX" = "cu100" ]; then
export CUDA_SHORT=10.0
export CUDA=10.0.130-1
export UBUNTU_VERSION=ubuntu1804
export CUBLAS=cuda-cublas-dev-10-0
export TOOLKIT="cudatoolkit=${CUDA_SHORT}"
fi
if [ "${TRAVIS_OS_NAME}" = "linux" ] && [ "$IDX" = "cu101" ]; then
export IDX=cu101
export CUDA_SHORT=10.1
export CUDA=10.1.105-1
export UBUNTU_VERSION=ubuntu1804
export CUBLAS=libcublas-dev
export TOOLKIT="cudatoolkit=${CUDA_SHORT}"
fi
if [ "${TRAVIS_OS_NAME}" = "osx" ]; then
export TOOLKIT=""
fi
if [ "${IDX}" = "cpu" ]; then
export FORCE_CPU=1
else
......
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