Перейти к содержимому
Compvision.ru
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

 

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

Поделиться сообщением


Ссылка на сообщение
Поделиться на других сайтах
2 hours ago, mrgloom said:

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

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

Поделиться сообщением


Ссылка на сообщение
Поделиться на других сайтах

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

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

 

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

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

 

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

 

 

 

Поделиться сообщением


Ссылка на сообщение
Поделиться на других сайтах

Попробовал 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 

 

Поделиться сообщением


Ссылка на сообщение
Поделиться на других сайтах

Создайте учётную запись или войдите для комментирования

Вы должны быть пользователем, чтобы оставить комментарий

Создать учётную запись

Зарегистрируйтесь для создания учётной записи. Это просто!

Зарегистрировать учётную запись

Войти

Уже зарегистрированы? Войдите здесь.

Войти сейчас


  • Сейчас на странице   0 пользователей

    Нет пользователей, просматривающих эту страницу

×