How to specify a particular GPU to train？

anosorae / IRRA

Cross-Modal Implicit Relation Reasoning and Aligning for Text-to-Image Person Retrieval (CVPR 2023)

MIT License

194 stars 27 forks source link

Open ppsychoo opened 10 months ago

ppsychoo commented 10 months ago

Thanks for your contribution, as a newbie I would like to ask how to specify a certain GPU to train the model?

ThomasjonLi commented 10 months ago

Change number of this in the script : CUDA_VISIBLE_DEVICES=0