📌  相关文章
📜  RuntimeError: 'out' 的张量在 CPU 上,参数 #1 'self' 的张量在 CPU 上,但希望它们在 GPU 上 - Python 代码示例

📅  最后修改于: 2022-03-11 14:45:06.232000             🧑  作者: Mango

代码示例1
device = "cuda" if torch.cuda.is_available() else "cpu"
model.to(device)