我可以设置 Flux.jl 中内置的任何保护措施或选项，以免最终意外分配 GPU 上的所有可用内存？我在 Flux.jl 文档中没有看到任何类型的 GPU 选项：https ://fluxml.ai/Flux.jl/stable/gpu/
1 回答
0
现在，Flux.jl 使用 CUDA.jl 作为 GPU 的接口。CUDA.jl 有一个名为的环境变量（您可以在文档中阅读）JULIA_CUDA_MEMORY_LIMIT，可用于：
限制分配的 GPU 内存量，以字节为单位。
注意这些环境变量应该在导入包之前设置好；在运行时更改它们没有任何效果。
您可以在此处更全面地了解 Flux 和 CUDA 中的内存管理：https ://cuda.juliagpu.org/stable/usage/memory/#Memory-management
于 2021-07-12T00:36:37.347 回答