import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" # see issue #152
os.environ["CUDA_VISIBLE_DEVICES"] = ""
本文介绍如何通过设置环境变量CUDA_DEVICE_ORDER和CUDA_VISIBLE_DEVICES来指定CUDA使用的GPU设备。这对于多GPU环境或者希望禁用某些GPU的情况下非常有用。
import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" # see issue #152
os.environ["CUDA_VISIBLE_DEVICES"] = ""
4656

被折叠的 条评论
为什么被折叠?
