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

前端 未结 3 550
故里飘歌
故里飘歌 2021-02-10 21:04

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:16

    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.

提交回复
热议问题