非常简单的 torch.tensor().to("cuda") 给出 CUDA 错误：设备端断言触发

Question

我所做的就是，

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
torch.tensor([123,123]).to(device)

我得到：

RuntimeError: CUDA error: device-side assert triggered

我真的不明白为什么。

编辑：奇怪的是，我发现这种情况只有在我事先运行一些代码后才会发生。重启内核即可解决。但为什么有些代码会导致这样的问题呢？我无法分享代码，因为它不是我的，但我仍然欢迎任何猜测。谢谢！

Answer 1

device 是一个字符串，而不是一个张量。将您的代码更改为

device = "cuda" if torch.cuda.is_available() else "cpu"

现在你的代码可以正常工作了。