Very simple torch.tensor().to(“cuda”) gives CUDA error: device-side assert triggered

前端 未结 0 1973
陌清茗
陌清茗 2020-12-01 15:36

All I do is,

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

And

相关标签:
回答
  • 消灭零回复
提交回复
热议问题