问:前向推理的时候 Yolov3 怎么指定 GPU?
来自社友的回答
▼▼▼
@邱忠喜:
os.environ[“CUDA_VISIBLE_DEVICES“]=INDEX index用GPUid代替id从前0开始, 也可在python文件中指定用os.environ[“CUDA_DEVICE_ORDER“]=“PCI_BUS_ID“] os.environ[“-, 其中index用可以用 nvidia-smi查看比如0, CUDA_VISIBLE_DEVICES=index python xxx.py, 使用cuda可以指定可用的GPU可在运行时这样写。