RuntimeError: Invalid device string: 'cuda:0,1,2,3'
时间: 2024-05-17 22:12:13 浏览: 429
pytorch模型提示超出内存RuntimeError: CUDA out of memory.
5星 · 资源好评率100%
这个错误通常是由于在指定 PyTorch 模型运行设备时,使用了错误的格式。正确的格式应该是 `'cuda:0,1,2,3'`,也就是将逗号放在数字之间而不是数字和冒号之间。请检查你的代码,确保使用了正确的格式。如果问题仍然存在,可以尝试使用单个 GPU 进行计算,或者检查你的 PyTorch 版本是否与你的 GPU 驱动版本兼容。
阅读全文