import torch
torch.cuda.get_device_capability(0)
添加cmake options：
-DCMAKE_CUDA_ARCHITECTURES=86 -DCMAKE_CUDA_COMPILER=/usr/local/cuda-11.8/bin/nvcc
cmake工程出现“CMAKE_CUDA_ARCHITECTURES must be non-empty if set.“的解决方法_failed to detect a default cuda architecture-CSDN博客
)
