RuntimeError: CUDA error: invalid device ordinal

解决方法:

在代码中加入

os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "0,1,2"

亲测有效

你可能感兴趣的:(pytorch)