Set TORCH_CUDA_ARCH_LIST

openmm · Nov 17, 2023 · ed48fcb · ed48fcb
1 parent 2b7e61f
commit ed48fcb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/ci-cuda12.yml b/.github/workflows/ci-cuda12.yml
@@ -72,7 +72,7 @@ jobs:
               -DENABLE_CUDA=${{ matrix.enable_cuda }} \
               -DTorch_DIR=$(python -c 'import torch.utils; print(torch.utils.cmake_prefix_path)')/Torch \
               -DCMAKE_INSTALL_PREFIX=$CONDA_PREFIX \
-              -DCMAKE_CUDA_ARCHITECTURES=OFF
+              -DTORCH_CUDA_ARCH_LIST=8.9
         make install
 
     - name: Test