test cuda

.travis.yml

 language: shell
 
 os:
-  - linux
+  # - linux
   # - osx
   - windows
 
@@ -9,7 +9,7 @@ env:
   global:
     - CUDA_HOME=/usr/local/cuda
   jobs:
-    - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.8 IDX=cpu
+    # - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.8 IDX=cpu
     # - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.8 IDX=cu92
     # - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.8 IDX=cu100
     # - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.8 IDX=cu101

script/cuda.sh

@@ -82,7 +82,11 @@ if [ "${TRAVIS_OS_NAME}" = "windows" ] && [ "${IDX}" != "cpu" ]; then
   PowerShell -Command "Start-Process -FilePath \"cuda_${CUDA}_win10\" -ArgumentList \"-s nvcc_${CUDA_SHORT}\" -Wait -NoNewWindow"
   ls /c
   ls /c/Program\ Files\ \(x86\)
-  ls /c/Program\ Files\ \(x86\)/NVIDIA\ GPU\ Computing\ Toolkit
-  ls /c/Program\ Files\ \(x86\)/NVIDIA\ GPU\ Computing\ Toolkit/CUDA
-  ls /c/Program\ Files\ \(x86\)/NVIDIA\ GPU\ Computing\ Toolkit/CUDA/v9.2
+  # ls /c/Program\ Files\ \(x86\)/NVIDIA\ GPU\ Computing\ Toolkit
+  # ls /c/Program\ Files\ \(x86\)/NVIDIA\ GPU\ Computing\ Toolkit/CUDA
+  # ls /c/Program\ Files\ \(x86\)/NVIDIA\ GPU\ Computing\ Toolkit/CUDA/v9.2
+  ls /c/Program\ Files
+  # ls /c/Program\ Files/NVIDIA\ GPU\ Computing\ Toolkit
+  # ls /c/Program\ Files/NVIDIA\ GPU\ Computing\ Toolkit/CUDA
+  # ls /c/Program\ Files/NVIDIA\ GPU\ Computing\ Toolkit/CUDA/v9.2
 fi

script/rename_wheel.py

@@ -14,6 +14,6 @@ for wheel in wheels:
         continue
 
     names = name.split('-')
-    name = '-'.join(names[:-4] + [names[-4] + '%2B' + idx] + names[-3:])
+    name = '-'.join(names[:-5] + [names[-5] + '%2B' + idx] + names[-4:])
     new_wheel = osp.join(*wheel.split(osp.sep)[:-1], name)
     os.rename(wheel, new_wheel)