CUDA内存不足。已尝试分配196.00 MiB（GPU 0；6.00 GiB总容量；已分配372.64 MiB；3.87 GiB可用；PyTorch总共保留376.00 MiB）如果保留内存>>已分配内存，请尝试设置max_split_size_mb以避免碎片。请参阅内存管理和PYTORCH_CUDA_ALLOC_COFF的文档 上面不是说可用那么多嘛 为什么说内存不足呀 批量大小都32了 不能再小了