lxmert icon indicating copy to clipboard operation
lxmert copied to clipboard

Can not use multiply gpus for bottom-up feature extraction

Open Wangt-CN opened this issue 4 years ago • 6 comments

Hi, Thanks for authors great doker environment for bu feature extraction. However, I find it cannot use the multiply GPUs by using CUDA_VISIBLE_DEVICES=0,1,2,3 in docker? And I check the extract.py file, it seemed that it does not support the multiply gpu extracting?

Or am I missing anything? Thanks~

Wangt-CN avatar Mar 28 '20 12:03 Wangt-CN