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

前端 未结 3 542
故里飘歌
故里飘歌 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:37

    While building OpenBLAS, you have to set the flag USE_OPENMP = 1 to enable OpenMP support. Next set Caffe to use OpenBLAS in the Makefile.config, please export the number of threads you want to use during runtime by setting up OMP_NUM_THREADS=n where n is the number of threads you want.

提交回复
热议问题