How to config the number of GPUs used in re-id?

LCFractal / AIC21-MTMC

🏆The 1st place solution of track3 (City-Scale Multi-Camera Vehicle Tracking) in the NVIDIA AI City Challenge at CVPR 2021 Workshop.

MIT License

128 stars 35 forks source link

How to config the number of GPUs used in re-id? #11

Closed wanglu362 closed 2 years ago

wanglu362 commented 2 years ago

I ran your all pipeline on AIC, but the program crashed when fork eight GPU processes in re-id. I realized there're only two GPUs in my system. How should I configure the number of GPUs to be implemented? Thank you very much and regards!

wanglu362 commented 2 years ago

See, there's a global var NUM_PROCESS = 8 in extract_image_feat.py. Have altered this and hopefully it works.

sxy233666 commented 1 year ago

Did you solve the problem?