nvcc fatal : Unsupported gpu architecture ‘compute_‘

我是编译darknet ros时遇到的,错误原因是显卡算力不对应
在CMakeList中有如下:

  set(
    CUDA_NVCC_FLAGS
    ${CUDA_NVCC_FLAGS};
    -O3
    -gencode arch=compute_30,code=sm_30
    -gencode arch=compute_35,code=sm_35
    -gencode arch=compute_50,code=[sm_50,compute_50]
    -gencode arch=compute_52,code=[sm_52,compute_52]
    -gencode arch=compute_61,code=sm_61
    -gencode arch=compute_62,code=sm_62
  )

我的显卡是3060,查看算力对应,应该是8.6,把上面的改成:

  set(
    CUDA_NVCC_FLAGS
    ${CUDA_NVCC_FLAGS};
    -O3
    -gencode arch=compute_86,code=sm_86
  )

就好了.

你可能感兴趣的:(可能用到的小问题,深度学习,神经网络,pytorch)