cuda 架构设置

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博客