caffe 编译遇到 nvcc fatal : Unsupported gpu architecture ‘compute_75‘

cuda版本为9.0,GPU为1660 SUPPER

本应该安装cuda 10.2,但通过修改caffe源码的cuda.cmake可以实现编译

1.第59行

改为

set(__archs_names "Kepler" "Maxwell" "Pascal" "All" "Manual")

2.第75行

改为

 elseif(${CUDA_ARCH_NAME} STREQUAL "Pascal")
    set(__cuda_arch_bin "75")

3.第97行

caffe 编译遇到 nvcc fatal : Unsupported gpu architecture ‘compute_75‘_第1张图片

set(__cuda_arch_bin "60 61")

然后重新编译caffe

你可能感兴趣的:(caffe 编译遇到 nvcc fatal : Unsupported gpu architecture ‘compute_75‘)