超级会员免费看
在使用torch.view()时遇到RuntimeError，提示输入不连续。通过调用tensor.contiguous()可以尝试解决，但在某些情况下仍无法解决问题。参考StackOverflow上的讨论，该错误可能与CUDA的内存分配有关，cudaLimitMallocHeapSize控制着设备上malloc和free系统调用使用的堆大小。设置此限制必须在使用这些系统调用的内核启动之前完成，否则会返回cudaErrorInvalidValue。此限制仅适用于计算能力2.0及更高的设备。
当使用torch.view()时
出现 RuntimeError: invalid argument 1: input is not contiguous at /pytorch/torch/lib/TH/generic/THTensor.c:231
可以使用 tensor.contiguous() 解决
比如：
mask[idx, :].view(bat_size, 1) 变成 mask[idx, :].contiguous().view(bat_size, 1)
不能解决问题
https://stackoverflow.com/questions/13994799/cuda-invalid-argument-error-on-second-kernel
cudaLimitMallocHeapSize controls the size in bytes of the heap used by the malloc() and free() device system calls. Setting cudaLimitMallocHeapSize must be performed before launching any kernel that uses the mallo
订阅专栏 解锁全文
4925
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
