How to use multi CPU cores to train NNs using caffe and OpenBLAS

前端 未结 3 1906
我在风中等你
我在风中等你 2021-02-10 20:57

I am learning deep learning recently and my friend recommended me caffe. After install it with OpenBLAS, I followed the tutorial, MNIST task in the doc. But later I found it was

3条回答
  •  余生分开走
    2021-02-10 21:36

    I found that this method works:

    When you build the caffe, in your make command, do use this for 8 cores: make all -j8 and make pycaffe -j8

    Also, make sure OPENBLAS_NUM_THREADS=8 is set.

    This question has a full script for the same.

提交回复
热议问题