我所做的就是,
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
torch.tensor([123,123]).to(device)
我得到:
RuntimeError: CUDA error: device-side assert triggered
我真的不明白为什么。
编辑:奇怪的是,我发现这种情况只有在我事先运行一些代码之后才会发生。重启内核即可解决。但为什么有些代码会导致这样的问题呢?我无法分享代码,因为它不是我的,但我仍然欢迎任何猜测。谢谢!