我有时会在训练模特的时候从记忆中恢复
我有时会在训练模特的时候从记忆中恢复
提问于 2020-05-23 17:51:41
Tensorflow版本: TF1.13使用Anaconda
我研究了堆栈溢出设置：
TF_CUDNN_WORKSPACE_LIMIT_IN_MB = 100以减少tensorflow的划痕空间，默认情况下为4GB
GPU : NVIDIA GTX 1660 TI 6GB卡CUDA版本: 11.0
但我不知道如何设置环境变量。我找不到任何关于它的教程。既然我是这个的初学者，有人能给出任何链接或告诉如何设置这个变量吗？？这对我很有帮助。
回答 1
Stack Overflow用户
回答已采纳
发布于 2020-05-25 03:50:25
您可以在python中使用OS模块，如下所示：
import os
os.environ['TF_CUDNN_WORKSPACE_LIMIT_IN_MB'] = '100'顺便说一句，100米B更不适合训练模特。至少分配1 1GiB。
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/61976288
复制相关文章
相似问题
腾讯云开发者
Copyright © 2013 - 2026 Tencent Cloud. All Rights Reserved. 腾讯云 版权所有
深圳市腾讯计算机系统有限公司 ICP备案/许可证号：粤B2-20090059 
粤公网安备44030502008569号
腾讯云计算（北京）有限责任公司 京ICP证150476号 | 京ICP备11018762号