tensorflow:指定gpu 限制使用量百分比,设置最小使用量的实现
更新时间：2020年02月06日 15:23:10 作者：ming.zhang
今天小编就为大家分享一篇tensorflow:指定gpu 限制使用量百分比,设置最小使用量的实现，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
在Python代码中指定GPU
import os os.environ["CUDA_VISIBLE_DEVICES"] = "0"
设置定量的GPU使用量：
config = tf.ConfigProto() config.gpu_options.per_process_gpu_memory_fraction = 0.9 # 占用GPU90%的显存 session = tf.Session(config=config)
设置最小的GPU使用量：
config = tf.ConfigProto() config.gpu_options.allow_growth = True session = tf.Session(config=config)
以上这篇tensorflow:指定gpu 限制使用量百分比,设置最小使用量的实现就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
浅谈spring boot 集成 log4j 解决与logback冲突的问题
今天小编就为大家分享一篇浅谈spring boot 集成 log4j 解决与logback冲突的问题，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-02-02
这篇文章主要介绍了Python格式化输出--%s,%d,%f的代码详解，本文通过实例代码给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2020-04-04
Python获取暗黑破坏神3战网前1000命位玩家的英雄技能统计
这篇文章主要介绍了Python获取暗黑3战网前1000命位玩家的英雄技能统计的方法,借助urllib2模块以类似爬虫的机制来实现,需要的朋友可以参考下2016-07-07
这篇文章主要介绍了Python复制文件操作的方法,涉及Python针对文件与目录的复制及删除操作相关技巧,具有一定参考借鉴价值,需要的朋友可以参考下2015-11-11
这篇文章主要介绍了Python中IPYTHON的用法入门,较为详细的分析了IPYTHON的配置及使用技巧,具有一定参考借鉴价值,需要的朋友可以参考下2015-05-05
这篇文章主要为大家详细介绍了Python数据分析之如何利用matplotlib进行绘图，文中的示例代码讲解详细，感兴趣的小伙伴可以了解一下2022-09-09
这篇文章主要介绍了Django启动时找不到mysqlclient问题解决方案,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-11-11
python Multiprocessing.Pool进程池模块详解
multiprocessing模块提供了一个Process类来代表一个进程对象，multiprocessing模块像线程一样管理进程，这个是multiprocessing的核心，它与threading很相似，对多核CPU的利用率会比threading好的多2022-10-10
python 中Arduino串口传输数据到电脑并保存至excel表格
这篇文章主要介绍了python Arduino串口传输数据到电脑并保存至excel表格,本文给大家介绍的非常详细，具有一定的参考借鉴价值，需要的朋友可以参考下2019-10-10
这篇文章主要介绍了Python Django框架防御CSRF攻击的方法,结合实例形式分析了Python Django框架防御CSRF攻击的原理、配置方法与使用技巧,需要的朋友可以参考下2019-10-10
最新评论