O Ubuntu não pode executar o modelo caffe no modo multigpu

Question

Eu tenho três GTX 1080 Ti conectados e estou tentando treinar um modelo com a opção multigpu. Mas isso não funciona. Com single gpu, treina bem mas com multigpu diz assim e não faz nada:

I1210 13:24:25.852613  3033 net.cpp:1358] [1] Reserving 582912 bytes of shared learnable space
I1210 13:24:25.852761  3032 net.cpp:1358] [0] Reserving 582912 bytes of shared learnable space
I1210 13:24:25.852787  3034 net.cpp:1358] [2] Reserving 582912 bytes of shared learnable space

Eu esperei o tempo suficiente, então eu tenho que matar o processo. Ele usa 100% de todos os gpus após este comando. O que devo fazer?

    

Answer 1

Nos sistemas AMD, ao tentar configurar o Ubuntu, você precisa ignorar as GPUs. Se você tentar treinar um modelo com essa seleção, receberá esse erro.

sudo nano / etc / defaults / grub

GRUB_CMDLINE_LINUX="iomnu = soft"

sudo update grub