小编roc*_*lls的帖子

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

我所做的就是,

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
torch.tensor([123,123]).to(device)
Run Code Online (Sandbox Code Playgroud)

我得到:

RuntimeError: CUDA error: device-side assert triggered
Run Code Online (Sandbox Code Playgroud)

我真的不明白为什么。

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

pytorch

7
推荐指数
1
解决办法
4384
查看次数

标签 统计

pytorch ×1