RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0

原创已于 2022-08-15 10:05:52 修改 · 2.3k 阅读

本内容遵循CC 4.0 BY-SA版权协议

于 2022-08-15 09:53:12 首次发布

博客主要讲述成功解决RuntimeError报错问题，该错误是在模型训练时，数据不在同一块GPU上导致。解决方案是在模型训练的代码最前面添加特定代码。

成功解决：RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cuda:1! (when checking argument for argument index in method wrapper__index_select)，报错问题。
在这里插入图片描述
问题原因：模型训练时，数据不在同一块GPU上
解决方案：在模型训练的代码最前面加上下面的代码即可

import os
# 指定使用一块GPU，[0,1,2,3]下面是只选择使用1号GPU
os.environ["CUDA_VISIBLE_DEVICES"] = "1"

标签

#深度学习 #python #人工智能