File tree Expand file tree Collapse file tree 2 files changed +12
-5
lines changed Expand file tree Collapse file tree 2 files changed +12
-5
lines changed Original file line number Diff line number Diff line change 2727 # - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.5 IDX=cu101
2828
2929jobs :
30- exclude : # Exclude all macOS CUDA jobs.
30+ exclude : # Exclude all macOS CUDA jobs and Windows cu100 job .
3131 - os : osx
3232 env : TORCH_VERSION=1.4.0 PYTHON_VERSION=3.8 IDX=cu92
3333 - os : osx
@@ -52,13 +52,22 @@ jobs:
5252 env : TORCH_VERSION=1.4.0 PYTHON_VERSION=3.5 IDX=cu100
5353 - os : osx
5454 env : TORCH_VERSION=1.4.0 PYTHON_VERSION=3.5 IDX=cu101
55+ - os : windows
56+ env : TORCH_VERSION=1.4.0 PYTHON_VERSION=3.8 IDX=cu100
57+ - os : windows
58+ env : TORCH_VERSION=1.4.0 PYTHON_VERSION=3.7 IDX=cu100
59+ - os : windows
60+ env : TORCH_VERSION=1.4.0 PYTHON_VERSION=3.6 IDX=cu100
61+ - os : windows
62+ env : TORCH_VERSION=1.4.0 PYTHON_VERSION=3.5 IDX=cu100
63+
5564
5665install :
5766 - source script/cuda.sh
5867 - source script/conda.sh
5968 - conda create --yes -n test python=${PYTHON_VERSION}
6069 - source activate test
61- - if [ ${TRAVIS_OS_NAME} = "windows" ] || [ $IDX != "cu92" ]; then conda install pytorch ${TOOLKIT} -c pytorch -c defaults -c numba/label/dev --yes; else conda install pytorch=${TORCH_VERSION} ${TOOLKIT} -c pytorch --yes; fi
70+ - if [ ${TRAVIS_OS_NAME} = "windows" ] || [ $IDX != "cu92" ]; then conda install pytorch=${TORCH_VERSION} ${TOOLKIT} -c pytorch -c defaults -c numba/label/dev --yes; else conda install pytorch=${TORCH_VERSION} ${TOOLKIT} -c pytorch --yes; fi
6271 - pip install flake8 codecov
6372 - python setup.py install
6473
Original file line number Diff line number Diff line change @@ -38,8 +38,6 @@ if [ "${TRAVIS_OS_NAME}" = "windows" ] && [ "$IDX" = "cu92" ]; then
3838 export CUDA_URL=https://developer.nvidia.com/compute/cuda/${CUDA_SHORT} /Prod2/local_installers2
3939 export CUDA_FILE=cuda_${CUDA_SHORT} .148_win10
4040 export TOOLKIT=" cudatoolkit=${CUDA_SHORT} "
41-
42-
4341fi
4442
4543if [ " ${TRAVIS_OS_NAME} " = " windows" ] && [ " $IDX " = " cu100" ]; then
8381
8482if [ " ${TRAVIS_OS_NAME} " = " windows" ] && [ " ${IDX} " != " cpu" ]; then
8583 wget -nv " ${CUDA_URL} /${CUDA_FILE} "
86- PowerShell -Command " Start-Process -FilePath \" ${CUDA_FILE} \" -ArgumentList \" -s nvcc_${CUDA_SHORT} cusparse_dev_${CUDA_SHORT} \" -Wait -NoNewWindow"
84+ PowerShell -Command " Start-Process -FilePath \" ${CUDA_FILE} \" -ArgumentList \" -s nvcc_${CUDA_SHORT} cublas_dev_ ${CUDA_SHORT} cusparse_dev_${CUDA_SHORT} \" -Wait -NoNewWindow"
8785 ls /c/Program\ Files\ \( x86\) /Microsoft\ Visual\ Studio/2017/BuildTools/VC/Tools/MSVC
8886 CUDA_HOME=/c/Program\ Files/NVIDIA\ GPU\ Computing\ Toolkit/CUDA/v${CUDA_SHORT}
8987 PATH=${CUDA_HOME} /bin:$PATH
You can’t perform that action at this time.
0 commit comments