fn6767

您用torch1.9跑没问题的cuda是cuda11吗？

我尝试了cuda11.1+torch1.9的配置，会报错malloc(): invalid size (unsorted)

或者报malloc(): memory corruption

应该是内存不够，我开始也碰到了这个问题，把batch_size改小或者加大内存