tensorflow1中如何设置GPU的使用
发布日期:2021-07-01 04:20:44 浏览次数:2 分类:技术文章

本文共 246 字,大约阅读时间需要 1 分钟。

import tensorflow as tf #tf.ConfigProto()主要的作用是配置tf.Session的运算方式config = tf.ConfigProto()config.gpu_options.allow_growth = True  # 不全部占满显存, 按需分配config.gpu_options.per_process_gpu_memory_fraction = 0.3  # 限制GPU内存占用率sess = tf.Session(config=config)

转载地址:https://mtyjkh.blog.csdn.net/article/details/114886577 如侵犯您的版权,请留言回复原文章的地址,我们会给您删除此文章,给您带来不便请您谅解!

上一篇:tensorflow2中如何设置GPU的使用
下一篇:如何在tensorflow2环境运行tensorflow1代码

发表评论

最新留言

路过按个爪印,很不错,赞一个!
[***.219.124.196]2024年04月28日 15时44分52秒