Tensorflow 版本：TF1.13，使用 Anaconda
我在 Stack Overflow 上学习到可以通过设置：
TF_CUDNN_WORKSPACE_LIMIT_IN_MB = 100来减少 Tensorflow 默认的4GB临时空间
GPU：NVIDIA GTX 1660 TI 6GB显卡CUDA 版本：11.0
但我不知道如何设置环境变量。我找不到任何相关的教程。作为一个初学者，有没有人能提供一些链接或告诉我如何设置这个变量？这对我会非常有帮助。
回答：
你可以使用 Python 中的 OS 模块，如下所示：
import osos.environ['TF_CUDNN_WORKSPACE_LIMIT_IN_MB'] = '100'顺便说一下，100MiB对于训练模型来说实在是太少了。至少应该分配1GiB。
