我有时会在训练模型时失去记忆

Question

张量流版本：TF1.13 使用 Anaconda

我研究了stack overflow来设置:
TF_CUDNN_WORKSPACE_LIMIT_IN_MB = 100 为默认为 4GB 的 tensorflow 减少划痕 space

GPU：NVIDIA GTX 1660 TI 6GB 显卡 CUDA 版本：11.0

但是不知道怎么设置环境变量。 我找不到任何关于它的教程。由于我是初学者，任何人都可以提供任何链接或告诉如何设置此变量吗？这对我真的很有帮助。

Answer 1

您可以在python中使用OS模块，如下所示：

import os

os.environ['TF_CUDNN_WORKSPACE_LIMIT_IN_MB'] = '100'

顺便说一句，100MiB 对于要训练的模型来说要少得多。至少给它分配 1GiB。