search

yuehuang2023 / cryoNeFEN

GNU General Public License v3.0
22 stars 0 forks source link

specify GPU id #10

Open DanfengSong opened 1 month ago

DanfengSong commented 1 month ago

How can I specify which GPU to use? Especially when GPU 0 is already occupied and unavailable?

yuehuang2023 commented 2 weeks ago

You can directly set the environment variable 'CUDA_VISIBLE_DEVICES' to a specific GPU ID.

DanfengSong commented 3 days ago

You can directly set the environment variable 'CUDA_VISIBLE_DEVICES' to a specific GPU ID.

Thank you for your response. I will try it. If there was a corresponding option for it, it would be great.