Computer vision 评估模型导致CUDA内存不足错误
我的模型接收一系列RGB图像（512 x 512 x 3），并基于此进行预测。我在测试224幅图像的序列时，出现了以下错误： 运行时错误：CUDA内存不足。尝试分配4.00 GiB（GPU 0；15.78 GiB总容量；10.21 GiB已分配；1.06 GiB空闲；PyTorch总共保留13.43 GiB） 这发生在Colab上，但我在AWS ml.p2.xlarge笔记本实例上也遇到了类似的错误 我看到的解决方案都指向减少批量大小，但这对我来说没有意义，因为我的批量大小已经是1: 火炬尺寸（[1224,3515112]） 如何解决此问题？看来我可能加载了一些错误的东西，需要4.00个Gig 512*512*3*224=176160768 编辑：还想补充一点，我试图将工人人数改为1人，但这并没有解决问题Computer vision 评估模型导致CUDA内存不足错误,computer-vision,pytorch,gpu,Computer Vision,Pytorch,Gpu,我的模型接收一系列RGB图像（512 x 512 x 3），并基于此进行预测。我在测试224幅图像的序列时，出现了以下错误： 运行时错误：CUDA内存不足。尝试分配4.00 GiB（GPU 0；15.78 GiB总容量；10.21 GiB已分配；1.06 GiB空闲；PyTorch总共保留13.43 GiB） 这发生在Colab上，但我在AWS ml.p2.xlarge笔记本实例上也遇到了类似的错误 我看到的解决方案都指向减少批量大小，但这对我来说没有意义，因为我的批量大小已经是1: 火炬尺寸（