tensorflow指定gpu但实际cpu运行_矩池云中Tensorflow指定GPU及GPU显存设置-白红宇的个人博客

tensorflow指定gpu但实际cpu运行_矩池云中Tensorflow指定GPU及GPU显存设置

发布日期：2021-06-24 17:05:40 浏览次数：2 分类：技术文章

本文共 799 字，大约阅读时间需要 2 分钟。

指定GPU

查看机器上GPU情况

命令： nvidia-smi

功能：显示机器上gpu的情况

命令： nvidia-smi -l

功能：定时更新显示机器上gpu的情况

其中左侧红框有0、1、2、3的编号，表示GPU的编号，在后面指定GPU时需要使用这个编号。

指定使用哪块GPU

~/ CUDA_VISIBLE_DEVICES=0  python your.py #使用GPU0~/ CUDA_VISIBLE_DEVICES=0,1 python your.py #使用GPU0,1# 注意单词不要打错 # 或者在程序开头os.environ['CUDA_VISIBLE_DEVICES'] = '0'  #使用 GPU 0os.environ['CUDA_VISIBLE_DEVICES'] = '0,1'  # 使用 GPU 0，1

限制GPU的使用率

方法一

config = tf.ConfigProto()config.gpu_options.per_process_gpu_memory_fraction = 0.85  #占用85%显存session = tf.Session(config=config)

方法二

gpu_options=tf.GPUOptions(per_process_gpu_memory_fraction=0.85)config=tf.ConfigProto(gpu_options=gpu_options)session = tf.Session(config=config)

参考文章

tf.ConfigProto和tf.GPUOptions用法总结

tensorflow学习笔记（二十五）：ConfigProto&GPU

tensorflow | ConfigProto GPU使用设置（包括占用率等）

tensorflow中指定GPU及GPU显存设置

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

上一篇：laravel mysql 事务_Laravel框架学习笔记之数据库事务

下一篇：printwriter 关闭后才提交写入?_图解MySQL | [原理解析] MySQL insert 语句的磁盘写入之旅...

发表评论

关于作者

喝酒易醉，品茶养心，人生如梦，品茶悟道，何以解忧？唯有杜康！

-- 愿君每日到此一游！

指定GPU

限制GPU的使用率

参考文章

发表评论

最新留言

关于作者

推荐文章