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

Question

Tensorflow版本：TF1.13使用Anaconda

我在堆栈溢出中研究了set：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。