liuwei16 / CSP

High-level Semantic Feature Detection: A New Perspective for Pedestrian Detection, CVPR, 2019
753 stars 191 forks source link

How to run test_wider_ms.py with multiple GPUs #82

Open nankaielsasong opened 4 years ago

nankaielsasong commented 4 years ago

Hi~

I ran the test_wider_ms.py with one GPU but it's too slow. Even if I set os.environ["CUDA_VISIBLE_DEVICES"] = "0, 1, 2", there is still only one working GPU, the memory of the other two GPUs are used, but the Volatile GPU-Util keeps 0% when I checked with command nvidia-smi. So can I run it with multiple GPUs to accelerate testing on the WiderFace dataset?

Thanks~