2023年12月19日
指定程序使用的GPU的设备号
摘要： 当主机上有多张GPU卡，可以通过如下方式设置代码main.py使用哪几张GPU卡来运行 import os os.environ['CUDA_DEVICE_ORDER'] = "PCI_BUS_ID" # 将cuda GPU调用顺序与nvidia-smi 显示顺序统一 os.environ['CUD 阅读全文
posted @ 2023-12-19 22:37 Bill_H 阅读(30) 评论(0) 推荐(0) 编辑