问如何防止Flux.jl分配我的GPU内存的全部？
EN

Stack Overflow用户

提问于 2021-06-28 23:03:51

回答 1查看 283关注 0票数 1

在Flux.jl中是否有任何可以设置的保护或选项，使我不会意外地在GPU上分配所有可用内存？我在Flux.jl文档中没有看到任何GPU选项：https://fluxml.ai/Flux.jl/stable/gpu/

回答已采纳

发布于 2021-07-12 00:36:37

现在，Flux.jl使用CUDA.jl作为GPU的接口。CUDA.jl有一个环境变量(可以是在文档里读到)，名为JULIA_CUDA_MEMORY_LIMIT，可用于：

限制分配的GPU内存量，以字节为单位。

请注意，这些环境变量应该在导入包之前设置；在运行时更改它们不会产生任何影响。

您可以在这里更广泛地阅读有关Flux和CUDA中的内存管理的内容：https://cuda.juliagpu.org/stable/usage/memory/#Memory-management

票数 0

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/68170868

复制

相似问题

问如何防止Flux.jl分配我的GPU内存的全部？EN