Update build.yml

ninele7 · Nov 4, 2022 · eae0659 · eae0659
1 parent 8b0650d
commit eae0659
Showing 1 changed file with 1 addition and 6 deletions.
diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
@@ -7,7 +7,7 @@ on:
 
 env:
   NVCC_FLAGS: "-allow-unsupported-compiler"
-  TORCH_CUDA_ARCH_LIST: "7.0;7.5;8.0;8.6"
+  TORCH_CUDA_ARCH_LIST: "7.0;7.5;8.0;8.6+PTX"
   DISTUTILS_USE_SDK: 1
   FORCE_CUDA: 1
 
@@ -23,11 +23,6 @@ jobs:
             python-version: '3.10'
             cuda-version: '11.3.1'
             torch-install-command: 'torch==1.12.1+cu113 torchvision==0.13.1+cu113 --extra-index-url https://download.pytorch.org/whl/cu113'
-          - os-version: ubuntu-latest
-            python-version: '3.10'
-            cuda-version: '11.7.0'
-            torch-install-command: 'torch==1.13.0+cu117 torchvision==0.14.0+cu117 --extra-index-url https://download.pytorch.org/whl/cu117'
-            custom-suffix: '+cu117'
           - os-version: windows-latest
             python-version: '3.10'
             cuda-version: '11.3.1'