If automatic GPU architecture detection fails, (as can happen if you have multiple GPUs installed), set the TCNN_CUDA_ARCHITECTURES environment variable for the GPU you would like to use. The following table lists the values for common GPUs. If your GPU is not listed, consult this exhaustive list.
原先设置如下:
-DCMAKE_CUDA_ARCHITECTURES=native -DCMAKE_CUDA_COMPILER="/usr/local/cuda-12.0/bin/nvcc"
修改后设置如下:
-DCMAKE_CUDA_ARCHITECTURES=86 -DCMAKE_CUDA_COMPILER="/usr/local/cuda-12.0/bin/nvcc"
参考下面的表格:
再次设置如下,即可: