different pythons

a372eb41 · rusty1s · 6067286b · a372eb41 · a372eb41
Commit a372eb41 authored Feb 12, 2020 by rusty1s
Show whitespace changes
Inline Side-by-side

Showing with 127 additions and 50 deletions

.travis.yml .travis.yml +94 -50

script/cuda.sh script/cuda.sh +33 -0

No files found.
--- a/.travis.yml
+++ b/.travis.yml
 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

--- a/script/install.sh
+++ b/script/install.sh
 #!/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