【发布时间】：2021-06-28 23:03:51
【问题描述】：
是否有任何内置于 Flux.jl 的保护措施或选项可以设置，这样我就不会意外地在我的 GPU 上分配所有可用内存？我在 Flux.jl 文档中没有看到任何类型的 GPU 选项：https://fluxml.ai/Flux.jl/stable/gpu/
【问题讨论】：
【发布时间】：2021-06-28 23:03:51
【问题描述】：
目前，Flux.jl 使用 CUDA.jl 作为 GPU 的接口。 CUDA.jl 有一个名为JULIA_CUDA_MEMORY_LIMIT 的环境变量（您可以使用read about in the docs），可用于：
限制分配的 GPU 内存量，以字节为单位。
注意这些环境变量要在导入包之前设置好；在运行时更改它们没有任何效果。
您可以在此处阅读有关 Flux 和 CUDA 内存管理的更多信息：https://cuda.juliagpu.org/stable/usage/memory/#Memory-management
【讨论】：