最大分辨率

您好, 我有兴趣直接使用您的推理代码进行图像到图像翻译。我想使用您的方法来编辑分辨率为 1920 x 1080 的图像。我使用的是 NVIDIA GeForce RTX 4090, 我相信它可以处理这种分辨率。然而, 每当我将分辨率设置为高于 900 时, 我都会遇到内存分配错误 (torch.cuda.OutOfMemoryError: CUDA out of memory)。此外, 运行代码时, 我还收到了以下警告: UserWarning: Failed to load the image Python extension: libtorch_cuda_cu.so: cannot open shared object file: No such file or directory。这有可能是相关的吗? 我很想知道您是否能根据提供的信息确定问题。我尝试简化模型(例如, 减少通道数、注意力分辨率等), 但质量下降过大。我还应该考虑其他策略吗?