在tensorflow中设置使用某一块GPU、多GPU、CPU的操作
更新时间：2020年02月07日 16:24:11 作者：MachineLP
今天小编就为大家分享一篇在tensorflow中设置使用某一块GPU、多GPU、CPU的操作，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
tensorflow下设置使用某一块GPU（从0开始编号）：
import os os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" os.environ["CUDA_VISIBLE_DEVICES"] = "1"
多GPU：
num_gpus = 4
for i in range(num_gpus):
with tf.device('/gpu:%d',%i):
。。。
只是用cpu的情况
with tf.device("/cpu:0"):
以上这篇在tensorflow中设置使用某一块GPU、多GPU、CPU的操作就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
这篇文章主要介绍了Python导出数据到Excel可读取的CSV文件的方法,设计Python操作Excel的相关技巧,需要的朋友可以参考下2015-05-05
Python PyQt5中弹出子窗口解决子窗口一闪而过的问题
这篇文章主要介绍了Python PyQt5中弹出子窗口解决子窗口一闪而过的问题，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2021-03-03
这篇文章主要介绍了Python 中 lru_cache 的使用和实现详解，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2021-01-01
Matplotlib是Python的第三方绘图库,它非常类似于MATLAB,在使用Matplotlib软件包之前需要对其进行安装,这篇文章主要给大家介绍了关于Matplotlib安装与配置的相关资料,需要的朋友可以参考下2023-09-09
这篇文章主要介绍了pytorch 把图片数据转化成tensor的操作，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2021-03-03
这篇文章主要介绍了python中的函数用法,包括了函数的定义及参数的各种注意事项等,对Python初学者有很好的借鉴价值,需要的朋友可以参考下2014-09-09
今天小编就为大家分享一篇Python 窗体(tkinter)按钮 位置实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-06-06
selector 是一个实现了IO复用模型的python包,实现了IO多路复用模型的 select、poll 和 epoll 等函数,下面就跟随小编一起来学习一下它的具体使用吧2024-02-02
这篇文章主要介绍了python Pandas时序数据处理，文章围绕主题展开详细的内容介绍，具有一定的参考价值，需要的小伙伴可以参考一下2022-06-06
这篇文章主要介绍了搞清楚 Python traceback的具体使用方法，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧2019-05-05
最新评论