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

Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐