Jump to content
Compvision.ru
Sign in to follow this  
mrgloom

Caffe multi CPU build

Recommended Posts

Собираю Caffe на Ubuntu c openblas в CPU_ONLY mode на Ubuntu 14.04 x64 которая крутится на VirtualBox.

https://github.com/mrgloom/Caffe-snippets/blob/master/Virtualbox_Ubuntu_14.04_CPU.sh

https://github.com/mrgloom/Caffe-snippets/blob/master/Makefile.config

 

https://github.com/xianyi/OpenBLAS

После выставление любой из переменных, я не наблюдаю ускорения вообще.

    export OPENBLAS_NUM_THREADS=4
    export GOTO_NUM_THREADS=4
    export OMP_NUM_THREADS=4

 

Как проверить что несколько потоков используется?

Share this post


Link to post
Share on other sites
2 hours ago, mrgloom said:

Как проверить что несколько потоков используется?

В top посмотреть на число создаваемых потоков с этими опциями выставлеными в 4 и в 1?

Share this post


Link to post
Share on other sites

Если я ставлю export OPENBLAS_NUM_THREADS=4

Потом захожу в top и нажимаю 'H', то сначала вроде появляется больше одного вхождения программы, но потом остаётся только 1 с загрузкой 100%.

 

top -H -p <PID> показывает 12 строчек как и ps -T -p <PID>

если использовать export OPENBLAS_NUM_THREADS=1 , то 9 строчек.

 

Треды создаются, но не используются?

 

 

 

Share this post


Link to post
Share on other sites

Попробовал caffe бинарник на Mnist http://caffe.berkeleyvision.org/gathered/examples/mnist.html

И тут уже 400% выдаёт

 

1 thread
I0520 15:58:09.749832 12424 caffe.cpp:178] Use CPU.
...
I0520 16:06:14.553506 12424 caffe.cpp:222] Optimization Done.
~8 min

4 threads
I0520 16:06:44.634735 12446 caffe.cpp:178] Use CPU.
...
I0520 16:13:15.904394 12446 caffe.cpp:222] Optimization Done.
~6.5 min

 

 

ps -T -p <PID>

1 thread 6 строчек

4 thread 9 строчек

Осталось понять когда мультипоточность задействуется и как это зависит от топологии сети.

И да по идее это должно ускорять и свёртки, т.к. всё сводится к вызову BLAS ф-ий.

http://caffe.berkeleyvision.org/tutorial/convolution.html 

 

Share this post


Link to post
Share on other sites

Create an account or sign in to comment

You need to be a member in order to leave a comment

Create an account

Sign up for a new account in our community. It's easy!

Register a new account

Sign in

Already have an account? Sign in here.

Sign In Now

Sign in to follow this  

  • Recently Browsing   0 members

    No registered users viewing this page.

×