How to assign certain GPUs？

boheumd / MA-LMM

(2024CVPR) MA-LMM: Memory-Augmented Large Multimodal Model for Long-Term Video Understanding

https://boheumd.github.io/MA-LMM/

MIT License

221 stars 26 forks source link

How to assign certain GPUs？ #19

Closed chhchan closed 3 months ago

chhchan commented 3 months ago

I directly use the yaml file you provide to train, and all of the 4 process use the GPUs randomly. How to modify the yaml file to let it use certain GPUs?

boheumd commented 3 months ago

You can set the cuda_visible_devices=0,1,2,3 for example to specify the certain GPUs you want to use.