📜  设置 cuda 可见设备 python 代码示例

📅  最后修改于: 2022-03-11 14:45:52.032000             🧑  作者: Mango

代码示例1
import os
os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"   # see issue #152
os.environ["CUDA_VISIBLE_DEVICES"]="0"