解决RuntimeError: CUDA error: invalid device ordinal

在程序中找到定义 device 的地方,把.cuda(output_device)中的属性值改为0,即意味着只有一个GPU训练模型

猜你喜欢

转载自blog.csdn.net/com_fang_bean/article/details/107528164