Skip to content

Commit 9bea677

Browse files
committed
remove cu100
1 parent d144cf1 commit 9bea677

File tree

2 files changed

+12
-5
lines changed

2 files changed

+12
-5
lines changed

.travis.yml

Lines changed: 11 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ env:
2727
# - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.5 IDX=cu101
2828

2929
jobs:
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

5665
install:
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

script/cuda.sh

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff 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-
4341
fi
4442

4543
if [ "${TRAVIS_OS_NAME}" = "windows" ] && [ "$IDX" = "cu100" ]; then
@@ -83,7 +81,7 @@ fi
8381

8482
if [ "${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

0 commit comments

Comments
 (0)