【发布时间】：2021-02-06 00:37:08
【问题描述】：
您好，我刚购买了新的 RTX 3080 GPU。我想将 Cuda 用于我的张量流模型。但是当我开始训练模型时，需要一个多小时才能开始训练。所以我去了 tensorflow 网站，那里说要增加 CUDA_CACHE_MAXSIZE 但我不知道那在哪里。有人可以帮我找到它并使尺寸更大吗？我的 GPU 是 10GB。谢谢！
【问题讨论】：
-
这是environment variable。设置环境变量的方法取决于您的操作系统（您没有指出），但无论如何设置环境变量的说明都很容易通过谷歌发现。您还可以通过使用为 Ampere GPU 编译的 tensorflow 版本来“修复”此问题，例如现在在 NGC 上提供的版本
-
您好，我使用的是 Windows 10，我有 TF 2.3.1。当我转到环境变量时，没有 cuda_cache_maxsize 我想创建一个带有路径的新环境变量吗？我在哪里可以找到 cuda_cache_maxsize 的路径？顺便说一句，我用 anaconda 安装了所有的 cuda 东西。谢谢！
-
是的，你需要create a new environment variable 没有与之关联的路径。您根据我给您的第一个链接中的说明将环境变量设置为等于一个值。而且，顺便说一句，如果设置此变量不会显着改善您的启动体验，我不会感到惊讶。但是试一试。
-
好的，谢谢！！！
标签： python tensorflow deep-learning nvidia