在程序中找到定义 device 的地方,把.cuda(output_device)中的属性值改为0,即意味着只有一个GPU训练模型
解决RuntimeError: CUDA error: invalid device ordinal
最新推荐文章于 2024-06-22 17:22:33 发布
在程序中找到定义 device 的地方,把.cuda(output_device)中的属性值改为0,即意味着只有一个GPU训练模型