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