Commit 4e79c0db authored by rusty1s's avatar rusty1s
Browse files

[ci skip]

parent 8e12a57b
...@@ -13,7 +13,7 @@ jobs: ...@@ -13,7 +13,7 @@ jobs:
os: [windows-latest] # [ubuntu-18.04, macos-10.15, windows-latest] os: [windows-latest] # [ubuntu-18.04, macos-10.15, windows-latest]
python-version: [3.6, 3.7, 3.8, 3.9] python-version: [3.6, 3.7, 3.8, 3.9]
torch-version: [1.8.0, 1.9.0] torch-version: [1.8.0, 1.9.0]
cuda-version: ['cpu', 'cu101', 'cu102', 'cu111'] cuda-version: ['cu111'] # ['cpu', 'cu101', 'cu102', 'cu111']
exclude: exclude:
- torch-version: 1.9.0 - torch-version: 1.9.0
cuda-version: 'cu101' cuda-version: 'cu101'
......
...@@ -5,4 +5,4 @@ PATH=${CUDA_HOME}/bin:$PATH ...@@ -5,4 +5,4 @@ PATH=${CUDA_HOME}/bin:$PATH
PATH=/c/Program\ Files\ \(x86\)/Microsoft\ Visual\ Studio/2017/BuildTools/MSBuild/15.0/Bin:$PATH PATH=/c/Program\ Files\ \(x86\)/Microsoft\ Visual\ Studio/2017/BuildTools/MSBuild/15.0/Bin:$PATH
export FORCE_CUDA=1 export FORCE_CUDA=1
export TORCH_CUDA_ARCH_LIST="3.5;5.0+PTX;6.0;7.0;7.5" export TORCH_CUDA_ARCH_LIST="3.5;5.0+PTX;6.0;7.0;7.5;8.0;8.6"
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment