PYTORCH_CUDA_ALLOC_CONF 中可扩展段的平台支持

Question

使用 torch 的应用程序不断告诉我“此平台不支持可扩展段”（即 EasyDiffusion 和 EasyTraining GUI），但我找不到任何有关此功能支持的信息。它需要特定的硬件或操作系统吗？我真的可以使用它应该提供的内存节省。

我在 Win10x64 和 RTX 2070S 8Gb、torch-2.3.1+cu121、torchvision-0.18.1+cu121 上运行 Cuda 12.1。我听说支持此功能的最后一个 pytorch 版本是 2.1.2，但尝试从 2.0.0 到 2.2.1 的版本却一无所获。

Answer 1

告诉你 Windows 不支持它会杀了你吗？