[ci skip]

Rejoker456 · Oct 22, 2021 · 8d05f61 · 8d05f61
1 parent 154e81b
commit 8d05f61
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/.github/workflows/building-conda.yml b/.github/workflows/building-conda.yml
@@ -10,10 +10,10 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        os: [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]
         torch-version: [1.9.0, 1.10.0]
-        cuda-version: ['cpu', 'cu102', 'cu111', 'cu113']
+        cuda-version: ['cu113'] #['cpu', 'cu102', 'cu111', 'cu113']
         exclude:
           - torch-version: 1.9.0
             cuda-version: 'cu113'

diff --git a/.github/workflows/cuda/cu113-Windows-env.sh b/.github/workflows/cuda/cu113-Windows-env.sh
@@ -5,4 +5,4 @@ PATH=${CUDA_HOME}/bin:$PATH
 PATH=/c/Program\ Files\ \(x86\)/Microsoft\ Visual\ Studio/2017/BuildTools/MSBuild/15.0/Bin:$PATH
 
 export FORCE_CUDA=1
-export TORCH_CUDA_ARCH_LIST="3.5;5.0+PTX;6.0;7.0;7.5;8.0;8.6"
+export TORCH_CUDA_ARCH_LIST="6.0+PTX"