conda python

.travis.yml

-sudo: enabled
-
 env:
   global:
     - TORCH_VERSION=1.4.0
@@ -12,33 +10,27 @@ jobs:
   include:
     - name: linux/cpu
       os: linux
-      language: python
-      python: 3.7.6
       env:
-        - IDX=cpu
-        - TORCH=${TORCH_VERSION}+${IDX}
+        - IDX = cpu
+        - TOOLKIT=cpuonly
     - name: linux/cu92
       os: linux
-      language: python
-      python: 3.7.6
       env:
         - IDX=cu92
         - CUDA_SHORT=9.2
         - CUDA=9.2.148-1
         - UBUNTU_VERSION=ubuntu1604
         - CUBLAS=cuda-cublas-dev-9-2
-        - TORCH=${TORCH_VERSION}+${IDX}
+        - TOOLKIT="cudatoolkit=${CUDA_SHORT}"
     - name: linux/cu100
       os: linux
-      language: python
-      python: 3.7.6
       env:
         - IDX=cu100
         - CUDA_SHORT=10.0
         - CUDA=10.0.130-1
         - UBUNTU_VERSION=ubuntu1804
         - CUBLAS=cuda-cublas-dev-10-0
-        - TORCH=${TORCH_VERSION}+${IDX}
+        - TOOLKIT="cudatoolkit=${CUDA_SHORT}"
     - name: linux/cu101
       os: linux
       language: python
@@ -49,36 +41,36 @@ jobs:
         - CUDA=10.1.105-1
         - UBUNTU_VERSION=ubuntu1804
         - CUBLAS=libcublas-dev
-        - TORCH=${TORCH_VERSION}
-    - name: macosx/cpu
-      os: osx
-      osx_image: xcode11.3
-      language: shell
-      env:
-        - IDX=cpu
-        - TORCH=${TORCH_VERSION}
-    - name: windows/cpu
-      os: windows
-      language: shell
-      env:
-        - IDX=cpu
-        - TORCH=${TORCH_VERSION}+${IDX}
-        - PATH=/c/Python37:/c/Python37/Scripts:$PATH
-      before_install:
-        - choco install python --version 3.7.6
-        - python -m pip install --upgrade --user pip
-    - name: windows/cu92
-      os: windows
-      language: shell
-      env:
-        - IDX=cu92
-        - CUDA_SHORT=9.2
-        - CUDA=9.2.148
-        - TORCH=${TORCH_VERSION}+${IDX}
-        - PATH=/c/Python37:/c/Python37/Scripts:$PATH
-      before_install:
-        - choco install python --version 3.7.6
-        - python -m pip install --upgrade --user pip
+        - TOOLKIT="cudatoolkit=${CUDA_SHORT}"
+    # - name: macosx/cpu
+    #   os: osx
+    #   osx_image: xcode11.3
+    #   language: shell
+    #   env:
+    #     - IDX=cpu
+    #     - TORCH=${TORCH_VERSION}
+    # - name: windows/cpu
+    #   os: windows
+    #   language: shell
+    #   env:
+    #     - IDX=cpu
+    #     - TORCH=${TORCH_VERSION}+${IDX}
+    #     - PATH=/c/Python37:/c/Python37/Scripts:$PATH
+    #   before_install:
+    #     - choco install python --version 3.7.6
+    #     - python -m pip install --upgrade --user pip
+    # - name: windows/cu92
+    #   os: windows
+    #   language: shell
+    #   env:
+    #     - IDX=cu92
+    #     - CUDA_SHORT=9.2
+    #     - CUDA=9.2.148
+    #     - TORCH=${TORCH_VERSION}+${IDX}
+    #     - PATH=/c/Python37:/c/Python37/Scripts:$PATH
+    #   before_install:
+    #     - choco install python --version 3.7.6
+    #     - python -m pip install --upgrade --user pip
     # - name: windows/cu100
     #   os: windows
     #   language: shell
@@ -103,35 +95,38 @@ jobs:
     #     - choco install python --version 3.7.6
 
 install:
-  - pip3 install --upgrade pip
-  - pip3 install wheel
   - source script/install.sh
-  - pip3 install numpy
-  - pip3 install torch==${TORCH} -f https://download.pytorch.org/whl/torch_stable.html
-  - pip3 install flake8
-  - pip3 install codecov
-  - python3 setup.py install || python setup.py install
+  - source script/conda.sh
+  - conda create -yes -n test python=3.7.6
+  - source activate test
+  - conda install pytorch=${TORCH_VERSION} {$TOOLKIT} -c pytorch
+  - python setup.py install
+  # - pip3 install --upgrade pip
+  # - pip3 install wheel
+  # - pip3 install numpy
+  # - pip3 install torch==${TORCH} -f https://download.pytorch.org/whl/torch_stable.html
+  # - pip3 install flake8
+  # - pip3 install codecov
 
-script:
-  - flake8 .
-  - python3 setup.py test || python setup.py test
-after_success:
-  - if [ "${TRAVIS_OS_NAME}" != "windows" ]; then python3 setup.py bdist_wheel --dist-dir=dist/${IDX}; fi
-  - if [ "${TRAVIS_OS_NAME}" = "windows" ]; then python setup.py bdist_wheel --dist-dir=dist/${IDX}; fi
-  - codecov
-deploy:
-  provider: s3
-  edge: true
-  skip_cleanup: true
-  access_key_id: AKIAJB7S6NJ5OM5MAAGA
-  secret_access_key: ${S3_SECRET_ACCESS_KEY}
-  bucket: pytorch-scatter
-  local-dir: dist/${IDX}
-  upload-dir: whl/${IDX}
-  acl: public_read
-  on:
-    repo: rusty1s/pytorch_scatter
-    # tags: true
-    branch: travis
+# script:
+#   - flake8 .
+#   - python3 setup.py test || python setup.py test
+# after_success:
+#   - if [ "${TRAVIS_OS_NAME}" != "windows" ]; then python3 setup.py bdist_wheel --dist-dir=dist/${IDX}; fi
+#   - if [ "${TRAVIS_OS_NAME}" = "windows" ]; then python setup.py bdist_wheel --dist-dir=dist/${IDX}; fi
+#   - codecov
+# deploy:
+#   provider: s3
+#   edge: true
+#   access_key_id: AKIAJB7S6NJ5OM5MAAGA
+#   secret_access_key: ${S3_SECRET_ACCESS_KEY}
+#   bucket: pytorch-scatter
+#   local_dir: dist/${IDX}
+#   upload_dir: whl/${IDX}
+#   acl: public_read
+#   on:
+#     repo: rusty1s/pytorch_scatter
+#     # tags: true
+#     branch: travis
 notifications:
   email: false

script/conda.sh

+#!/bin/bash
+
+if [ "${TRAVIS_OS_NAME}" = "linux" ]; then
+  wget http://repo.continuum.io/miniconda/Miniconda-latest-Linux-x86_64.sh -O miniconda.sh
+  chmod +x miniconda.sh
+  ./miniconda.sh -b
+  PATH=/home/travis/miniconda/bin:${PATH}
+  conda update --yes conda
+fi
+
+

script/install.sh

@@ -21,5 +21,5 @@ fi
 
 if [ "${TRAVIS_OS_NAME}" = "windows" ] && [ "${IDX}" != "cpu" ]; then
   wget "https://developer.nvidia.com/compute/cuda/${CUDA_SHORT}/Prod2/local_installers2/cuda_${CUDA}_win10"
-  ./cuda_${CUDA}_win10
+  # ./cuda_${CUDA}_win10
 fi