此环境变量用于控制是否关闭内存复用机制。
此环境变量默认为未配置。
关闭内存复用机制后,每次申请内存通过aclrtmalloc接口,生命周期结束后,立即通过aclrtfree接口释放回驱动。
关闭内存复用机制示例:
export PYTORCH_NO_NPU_MEMORY_CACHING=1
重新启用内存复用机制示例:
unset PYTORCH_NO_NPU_MEMORY_CACHING
否
此环境变量仅适用于PyTorch网络,且使用pytorch内存机制管理内存的场景。
Atlas 训练系列产品
Atlas A2 训练系列产品