search

Stinky-Tofu / Stronger-yolo

🔥Improve yolo with latest paper
MIT License
3 stars 0 forks source link

怎么在训练的时候指定1卡GPU？ #45

Open gooodooo opened 5 years ago

gooodooo commented 5 years ago

我有两张卡，不想用0卡，只想用1卡，请问一下该怎么设置

polar99 commented 5 years ago

Add the following code to train.py import os os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" os.environ['CUDA_VISIBLE_DEVICES'] = "1"