windows cuda

.travis.yml

@@ -62,7 +62,16 @@ jobs:
       env:
         - TORCH=${TORCH_VERSION}+cpu
         - PATH=/c/Python37:/c/Python37/Scripts:$PATH
-        - EXTRA_COMPILE_ARGS="-IC:\\Program Files (x86)\\Microsoft Visual Studio\\2017\\include"
+        # - EXTRA_COMPILE_ARGS="-IC:\\Program Files (x86)\\Microsoft Visual Studio\\2017\\include"
+      before_install:
+        - choco install python --version 3.7.6
+    - name: windows/cu101
+      os: windows
+      language: shell  # `language: python` is an error on Travis CI Windows
+      env:
+        - TORCH=${TORCH_VERSION}
+        - PATH=/c/Python37:/c/Python37/Scripts:$PATH
+        # - EXTRA_COMPILE_ARGS="-IC:\\Program Files (x86)\\Microsoft Visual Studio\\2017\\include"
       before_install:
         - choco install python --version 3.7.6
 

install.sh

@@ -14,3 +14,8 @@ if [ "${TRAVIS_OS_NAME}" = "linux" ] && [ "${FORCE_CUDA}" == "1" ]; then
   PATH=${CUDA_HOME}/bin:${PATH}
   nvcc --version
 fi
+
+if [ "${TRAVIS_OS_NAME}" = "windows" ] && [ "${FORCE_CUDA}" == "1" ]; then
+  wget -q http://developer.download.nvidia.com/compute/cuda/10.1/Prod/local_installers/cuda_10.1.243_426.00_win10.exe
+  ./cuda_10.1.243_426.00_win10.exe
+fi