微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

如何更改CUDA_CACHE_MAXSIZE NVIDIA

如何解决如何更改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)文档获取更多详细信息。

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。