hszhao / PSPNet

Pyramid Scene Parsing Network, CVPR2017.
https://hszhao.github.io/projects/pspnet
Other
1.59k stars 545 forks source link

Using several GPU to run the test code #72

Open JiYuanFeng opened 6 years ago

JiYuanFeng commented 6 years ago

hey! everything performs perfectly when I using singe GPU to implement the testing, However, when I try to make good use of my GPU resource, something dump. I use 3 GPU and set the step to 483,(1449/3). But the error was printed:

"""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""" Error using parallel_function(line 604) ALL workers aborted during execution of the parfor loop Error in sup_eval_all(line 71) parfor i =1:gpu_num

""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""

how should i solve this problem? Thanks guy~

pewpewpeww commented 6 years ago

Hi, I am also facing the same issue with 2 GPUs. May I know what is your code for single GPU? and how much memory is required for cityscapes data eval?

It's ok, i figured out I need to solve the same issue as you too since i have insufficient memory to run all cityscapes dataset with only 1 8GB GPU.