如何更改CUDA_CACHE_MAXSIZE NVIDIA

问题描述

您好,我刚刚购买了新的RTX 3080 GPU。我想将Cuda用于我的Tensorflow模型。但是,当我开始训练模型时,需要花费一个多小时来开始训练模型。所以我去了tensorflow网站,它说要增加CUDA_CACHE_MAXSIZE,但我不知道那在哪里。有人可以帮我找到它并扩大尺寸吗?我的GPU是10GB。谢谢!

解决方法

通过结合 Robert Crovella 在评论部分中提供的解决方案来证明答案中的解决方案。

CUDA_​CACHE_​MAXSIZE是“编译”类别下的一个环境变量。默认情况下为256 MB,最大为4 GB,您可以使用以下路径在Windows上设置此环境变量。

Control Panel ->System and Security->System->Advanced System settings 

您可以按照(12)文档获取更多详细信息。