在训练模型时，我有时会记忆力不足

Question

Tensorflow版本：TF1.13 使用水蟒

我研究了堆栈溢出来设置：
TF_CUDNN_WORKSPACE_LIMIT_IN_MB = 100 为了减少张量流的临时空间，默认情况下为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。