当前位置:  开发笔记 > 编程语言 > 正文

Caffe可以扩展到多个CPU核心吗?

如何解决《Caffe可以扩展到多个CPU核心吗?》经验,为你挑选了1个好方法。



1> avtomaton..:

通常caffe在其源代码中不支持多个CPU /核心,但它使用BLAS例程.

因此,您的问题的答案如下:

    是的,但只能通过BLAS配置,即您的BLAS版本应该使用多线程支持进行编译(参见相关讨论:此处或此处 - 在第二个链接中,您还可以找到对caffe本身的一些修改).

    同样通过BLAS(如果使用openmp支持编译,则可以定义OMP_NUM_THREADS为所需的值).

推荐阅读
有风吹过best
这个屌丝很懒,什么也没留下!
DevBox开发工具箱 | 专业的在线开发工具网站    京公网安备 11010802040832号  |  京ICP备19059560号-6
Copyright © 1998 - 2020 DevBox.CN. All Rights Reserved devBox.cn 开发工具箱 版权所有