windows try

.travis.yml

 language: shell
 
 os:
-  - linux
-  - osx
-  # - windows
+  # - linux
+  # - osx
+  - windows
 
 env:
   global:
@@ -11,12 +11,12 @@ env:
   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.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
@@ -53,85 +53,6 @@ jobs:
     - 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
-    #   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
-    #   env:
-    #     - FORCE_CUDA=1
-    #     - CUDA_SHORT=10.1
-    #     - CUDA=10.0.130_411.31
-    #     - TORCH=${TORCH_VERSION}+100
-    #     - PATH=/c/Python37:/c/Python37/Scripts:$PATH
-    #   before_install:
-    #     - choco install python --version 3.7.6
-    # - name: windows/cu102
-    #   os: windows
-    #   language: shell
-    #   env:
-    #     - FORCE_CUDA=1
-    #     - CUDA_SHORT=10.1
-    #     - CUDA=10.1.105_418.96
-    #     - TORCH=${TORCH_VERSION}
-    #     - PATH=/c/Python37:/c/Python37/Scripts:$PATH
-    #   before_install:
-    #     - choco install python --version 3.7.6
-
 install:
   - source script/cuda.sh
   - source script/conda.sh

script/cuda.sh

@@ -29,7 +29,29 @@ if [ "${TRAVIS_OS_NAME}" = "linux" ] && [ "$IDX" = "cu101" ]; then
   export TOOLKIT="cudatoolkit=${CUDA_SHORT}"
 fi
 
-if [ "${TRAVIS_OS_NAME}" = "osx" ]; then
+if [ "${TRAVIS_OS_NAME}" = "windows" ] && [ "$IDX" = "cpu" ]; then
+  export TOOLKIT=cpuonly
+fi
+
+if [ "${TRAVIS_OS_NAME}" = "windows" ] && [ "$IDX" = "cu92" ]; then
+  export CUDA_SHORT=9.2
+  export CUDA=9.2.148
+  export TOOLKIT="cudatoolkit=${CUDA_SHORT}"
+fi
+
+if [ "${TRAVIS_OS_NAME}" = "windows" ] && [ "$IDX" = "cu100" ]; then
+  export CUDA_SHORT=10.0
+  export CUDA=10.0.130_411.31
+  export TOOLKIT="cudatoolkit=${CUDA_SHORT}"
+fi
+
+if [ "${TRAVIS_OS_NAME}" = "windows" ] && [ "$IDX" = "cu101" ]; then
+  export CUDA_SHORT=10.1
+  export CUDA=10.1.105_418.96
+  export TOOLKIT="cudatoolkit=${CUDA_SHORT}"
+fi
+
+if [ "${TRAVIS_OS_NAME}" = "osx" ] && [ "$IDX" = "cpu" ]; then
   export TOOLKIT=""
 fi
 
@@ -54,7 +76,8 @@ if [ "${TRAVIS_OS_NAME}" = "linux" ] && [ "${IDX}" != "cpu" ]; then
   nvcc --version
 fi
 
-# if [ "${TRAVIS_OS_NAME}" = "windows" ] && [ "${IDX}" != "cpu" ]; then
-#   wget "https://developer.nvidia.com/compute/cuda/${CUDA_SHORT}/Prod2/local_installers2/cuda_${CUDA}_win10"
+if [ "${TRAVIS_OS_NAME}" = "windows" ] && [ "${IDX}" != "cpu" ]; then
+  wget -q "https://developer.nvidia.com/compute/cuda/${CUDA_SHORT}/Prod2/local_installers2/cuda_${CUDA}_win10"
+  ls
 #   # ./cuda_${CUDA}_win10
-# fi
+fi

script/wheel.py

@@ -3,18 +3,44 @@ import boto3
 s3_resource = boto3.resource('s3')
 bucket = s3_resource.Bucket(name="pytorch-scatter")
 objects = bucket.objects.all()
-wheels = sorted([obj.key[4:] for obj in objects if obj.key[-3:] == 'whl'])
+wheels = sorted([obj.key for obj in objects if obj.key[-3:] == 'whl'])
 
-content = '<!DOCTYPE html>\n<html>\n<body>\n{}\n</body>\n</html>'
-url = 'https://pytorch-scatter.s3.eu-central-1.amazonaws.com/whl/{}'
-links = ['<a href="{}">{}</a><br/>'.format(url.format(w), w) for w in wheels]
-content = content.format('\n'.join(links))
+wheels_dict = {}
+for torch_version in list(set([wheel.split('/')[1] for wheel in wheels])):
+    wheels_dict[torch_version] = []
+
+for wheel in wheels:
+    torch_version = wheel.split('/')[1]
+    wheels_dict[torch_version].append(wheel)
+
+html = '<!DOCTYPE html>\n<html>\n<body>\n{}\n</body>\n</html>'
+href = '<a href="{}">{}</a><br/>'
+
+url = 'http://pytorch-scatter.s3-website.eu-central-1.amazonaws.com/{}.html'
+index_html = html.format('\n'.join([
+    href.format(url.format('whl/' + key), key) for key in wheels_dict.keys()
+]))
 
 with open('index.html', 'w') as f:
-    f.write(content)
+    f.write(index_html)
 
 bucket.Object('whl/index.html').upload_file(
     Filename='index.html', ExtraArgs={
         'ContentType': 'text/html',
         'ACL': 'public-read'
     })
+
+url = 'https://pytorch-scatter.s3.eu-central-1.amazonaws.com/{}'
+for key, item in wheels_dict.items():
+    version_html = html.format('\n'.join([
+        href.format(url.format(i), '/'.join(i.split('/')[2:])) for i in item
+    ]))
+
+    with open('{}.html'.format(key), 'w') as f:
+        f.write(version_html)
+
+    bucket.Object('whl/{}.html'.format(key)).upload_file(
+        Filename='{}.html'.format(key), ExtraArgs={
+            'ContentType': 'text/html',
+            'ACL': 'public-read'
+        })