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

前端 未结 3 1903
我在风中等你
我在风中等你 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:18

    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.

提交回复
热议问题