Pytorch get“；运行时错误：CUDA错误：设备端断言已触发；_Pytorch

Pytorch get“；运行时错误：CUDA错误：设备端断言已触发；

pytorch

Pytorch get“；运行时错误：CUDA错误：设备端断言已触发；,pytorch,Pytorch,我的代码片段： max = torch.tensor([3]) if USE_CUDA: max = max.cuda() max_embedding = self.max_embedding(max) # dim of max_embedding: 1*5 item_dict = {} for item in item_list: item = torch.tensor(item) if USE_CUDA: item = item.cuda() item_embedd

我的代码片段：

max = torch.tensor([3])
if USE_CUDA: max = max.cuda()
max_embedding = self.max_embedding(max) # dim of max_embedding: 1*5

item_dict = {}
for item in item_list:
    item = torch.tensor(item)
    if USE_CUDA: item = item.cuda()
    item_embedding = self.item_embedding(item) # dim of item_embedding: 1*20

embedded = torch.cat((max_embedding, item_embedding), 1)

但我得到的错误是“RuntimeError:CUDA error:device side assert triggered”。通过添加

CUDA\u LAUNCH\u BLOCKING=1的输出：
/pytorch/aten/src/THC/THCTensorIndex.cu:308: void indexSelectSmallIndex(TensorInfo<T, IndexType>, TensorInfo<long, IndexType>, int, int, IndexType, long) [with T = float, IndexType = unsigned int, DstDim =2, SrcDim = 2, IdxDim = -2: block: [0,0,0], thread: [0,0,0] Assertion `srcIndex < srcSelectDimSize` failed.
/pytorch/aten/src/THC/THCTensorIndex.cu:308: void indexSelectSmallIndex(TensorInfo<T, IndexType>, TensorInfo<long, IndexType>, int, int, IndexType, long) [with T = float, IndexType = unsigned int, DstDim =2, SrcDim = 2, IdxDim = -2: block: [0,0,0], thread: [1,0,0] Assertion `srcIndex < srcSelectDimSize` failed.
/pytorch/aten/src/THC/THCTensorIndex.cu:308: void indexSelectSmallIndex(TensorInfo<T, IndexType>, TensorInfo<long, IndexType>, int, int, IndexType, long) [with T = float, IndexType = unsigned int, DstDim =2, SrcDim = 2, IdxDim = -2: block: [0,0,0], thread: [2,0,0] Assertion `srcIndex < srcSelectDimSize` failed.
/pytorch/aten/src/THC/THCTensorIndex.cu:308: void indexSelectSmallIndex(TensorInfo<T, IndexType>, TensorInfo<long, IndexType>, int, int, IndexType, long) [with T = float, IndexType = unsigned int, DstDim =2, SrcDim = 2, IdxDim = -2: block: [0,0,0], thread: [3,0,0] Assertion `srcIndex < srcSelectDimSize` failed.
/pytorch/aten/src/THC/THCTensorIndex.cu:308: void indexSelectSmallIndex(TensorInfo<T, IndexType>, TensorInfo<long, IndexType>, int, int, IndexType, long) [with T = float, IndexType = unsigned int, DstDim =2, SrcDim = 2, IdxDim = -2: block: [0,0,0], thread: [4,0,0] Assertion `srcIndex < srcSelectDimSize` failed.
Traceback (most recent call last):
  File "mytest.py", line 33, in <module>
     if USE_CUDA: item = item.cuda()
RuntimeError: CUDA error: device-side assert triggered

/pytorch/aten/src/THC/THCTensorIndex.cu:308:void indexSelectSmallIndex（TensorInfo，TensorInfo，int，int，IndexType，long）[带T=float，IndexType=unsigned int，DstDim=2，SrcDim=2，IdxDim=-2:block:[0,0,0]，thread:[0,0]断言'srcdex

如何修复它？
这是一个典型的索引越界错误案例，在嵌入的上下文中表现出来。请检查这一点，以找到类似问题的解决方案。
检查。这家伙也有同样的问题，因为他的一些标签有负值。你的GPU是什么？torch.version.cuda

的结果是什么？