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

Question

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

我研究的是堆栈溢出来设置：
TF_CUDNN_WORKSPACE_LIMIT_IN_MB = 100 减少 tensorflow 的暂存空间，默认为 4GB

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。