记录一个error--CUDA error: device-si

2020-07-28  本文已影响0人  asl_1da7

这个错误的关键在于错误并不在出错的哪一行。
看了一些网上的帖子和自己错误,错误的原因是用index超过了tensor的范围。
比如:

a = torch.randn(10)
a[11]

这里的11超过了a的范围。

上一篇 下一篇

猜你喜欢

热点阅读