|
35 | 35 |
|
36 | 36 | if [ "${TRAVIS_OS_NAME}" = "windows" ] && [ "$IDX" = "cu92" ]; then |
37 | 37 | export CUDA_SHORT=9.2 |
38 | | - export CUDA=9.2.148 |
| 38 | + export CUDA_URL=https://developer.nvidia.com/compute/cuda/${CUDA_SHORT}/Prod2/local_installers2 |
| 39 | + export CUDA_FILE=cuda_${CUDA_SHORT}.148_win10 |
39 | 40 | export TOOLKIT="cudatoolkit=${CUDA_SHORT}" |
| 41 | + |
| 42 | + |
40 | 43 | fi |
41 | 44 |
|
42 | 45 | if [ "${TRAVIS_OS_NAME}" = "windows" ] && [ "$IDX" = "cu100" ]; then |
43 | 46 | export CUDA_SHORT=10.0 |
44 | | - export CUDA=10.0.130_411.31 |
| 47 | + export CUDA_URL=https://developer.nvidia.com/compute/cuda/${CUDA_SHORT}/Prod/local_installers |
| 48 | + export CUDA_FILE=cuda_${CUDA_SHORT}.130_411.31_win10 |
45 | 49 | export TOOLKIT="cudatoolkit=${CUDA_SHORT}" |
46 | 50 | fi |
47 | 51 |
|
48 | 52 | if [ "${TRAVIS_OS_NAME}" = "windows" ] && [ "$IDX" = "cu101" ]; then |
49 | 53 | export CUDA_SHORT=10.1 |
50 | | - export CUDA=10.1.105_418.96 |
| 54 | + export CUDA_URL=https://developer.nvidia.com/compute/cuda/${CUDA_SHORT}/Prod/local_installers |
| 55 | + export CUDA_FILE=cuda_${CUDA_SHORT}.105_418.96_win10.exe |
51 | 56 | export TOOLKIT="cudatoolkit=${CUDA_SHORT}" |
52 | 57 | fi |
53 | 58 |
|
@@ -77,12 +82,9 @@ if [ "${TRAVIS_OS_NAME}" = "linux" ] && [ "${IDX}" != "cpu" ]; then |
77 | 82 | fi |
78 | 83 |
|
79 | 84 | if [ "${TRAVIS_OS_NAME}" = "windows" ] && [ "${IDX}" != "cpu" ]; then |
80 | | - wget -nv "https://developer.nvidia.com/compute/cuda/${CUDA_SHORT}/Prod2/local_installers2/cuda_${CUDA}_win10" |
81 | | - PowerShell -Command "Start-Process -FilePath \"cuda_${CUDA}_win10\" -ArgumentList \"-s nvcc_${CUDA_SHORT}\" -Wait -NoNewWindow" |
82 | | - ls /c/Program\ Files |
83 | | - ls /c/Program\ Files/NVIDIA\ GPU\ Computing\ Toolkit |
84 | | - ls /c/Program\ Files/NVIDIA\ GPU\ Computing\ Toolkit/CUDA |
85 | | - ls /c/Program\ Files/NVIDIA\ GPU\ Computing\ Toolkit/CUDA/v${CUDA_SHORT} |
| 85 | + wget -nv "${CUDA_URL}/${CUDA_FILE}" |
| 86 | + PowerShell -Command "Start-Process -FilePath \"${CUDA_FILE}\" -ArgumentList \"-s nvcc_${CUDA_SHORT}\" -Wait -NoNewWindow" |
| 87 | + ls /c/Program\ Files\ \(x86\)/Microsoft\ Visual\ Studio/2017 |
86 | 88 | CUDA_HOME=/c/Program\ Files/NVIDIA\ GPU\ Computing\ Toolkit/CUDA/v${CUDA_SHORT} |
87 | 89 | PATH=${CUDA_HOME}/bin:$PATH |
88 | 90 | PATH=/c/Program\ Files\ \(x86\)/Microsoft\ Visual\ Studio/2017/BuildTools/MSBuild/15.0/Bin:$PATH |
|
0 commit comments