OutOfMemoryError:设备0上的分配将超过允许的内存。（内存不足）当前分配：5.06 GiB请求：1.98 GiB设备限制：8.00 GiB免费（根据CUDA）：0字节PyTorch限制（由用户提供的内存分数设置）：17179869184.00 GiB
用时:9.0秒。
A： 5.06 GB，R:7.16 GB，系统：8.0/7.99609 GB（100.0%）