update

.github/workflows/building.yml

@@ -13,7 +13,7 @@ jobs:
         os: [ubuntu-16.04] #, macos-10.15, windows-latest]
         python-version: [3.6] #, 3.7, 3.8, 3.9]
         torch-version: [1.8.0] #, 1.9.0]
-        cuda-version: ['cpu', 'cu101'] #, 'cu102', 'cu111']
+        cuda-version: ['cpu', 'cu101', 'cu102', 'cu111']
         exclude:
           - torch-version: 1.9.0
             cuda-version: 'cu101'
@@ -56,7 +56,7 @@ jobs:
         if: ${{ matrix.cuda-version != 'cpu' }}
         run: |
           source .github/workflows/cuda/${{ matrix.cuda-version }}-${{ runner.os }}-env.sh
-          FORCE_CUDA=1 TORCH_CUDA_ARCH_LIST="5.2 6.0 6.1 7.0 7.5 8.0 8.6+PTX" pip install -e .
+          pip install -e .
         shell:
           bash
 

.github/workflows/cuda/cu101-Linux-env.sh

@@ -3,3 +3,6 @@
 CUDA_HOME=/usr/local/cuda-10.1
 LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
 PATH=${CUDA_HOME}/bin:${PATH}
+
+export FORCE_CUDA=1
+export TORCH_CUDA_ARCH_LIST="3.5;5.0+PTX;6.0;7.0;7.5"

.github/workflows/cuda/cu101-Windows-env.sh

@@ -3,3 +3,6 @@
 CUDA_HOME=/c/Program\ Files/NVIDIA\ GPU\ Computing\ Toolkit/CUDA/v10.1
 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"

.github/workflows/cuda/cu102-Linux-env.sh

@@ -3,3 +3,6 @@
 CUDA_HOME=/usr/local/cuda-10.2
 LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
 PATH=${CUDA_HOME}/bin:${PATH}
+
+export FORCE_CUDA=1
+export TORCH_CUDA_ARCH_LIST="3.5;5.0+PTX;6.0;7.0;7.5"

.github/workflows/cuda/cu102-Windows-env.sh

@@ -3,3 +3,6 @@
 CUDA_HOME=/c/Program\ Files/NVIDIA\ GPU\ Computing\ Toolkit/CUDA/v10.2
 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"

.github/workflows/cuda/cu111-Linux-env.sh

@@ -3,3 +3,6 @@
 CUDA_HOME=/usr/local/cuda-11.1
 LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
 PATH=${CUDA_HOME}/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"

.github/workflows/cuda/cu111-Windows-env.sh

@@ -3,3 +3,6 @@
 CUDA_HOME=/c/Program\ Files/NVIDIA\ GPU\ Computing\ Toolkit/CUDA/v11.1
 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"