在程序中找到定义 device 的地方,把.cuda(output_device)中的属性值改为0,即意味着只有一个GPU训练模型
解决RuntimeError: CUDA error: invalid device ordinal
猜你喜欢
转载自blog.csdn.net/com_fang_bean/article/details/107528164
今日推荐
周排行