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行
set(__cuda_arch_bin "60 61")
然后重新编译caffe