在tensorflow中设置使用某一块GPU、多GPU、CPU的操作


Posted in Python onFebruary 07, 2020

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 相关文章推荐
Python中字典的基本知识初步介绍
May 21 Python
Python 编码Basic Auth使用方法简单实例
May 25 Python
基于Python闭包及其作用域详解
Aug 28 Python
Django基于ORM操作数据库的方法详解
Mar 27 Python
理想高通滤波实现Python opencv示例
Jan 30 Python
python按照多个条件排序的方法
Feb 08 Python
详解如何设置Python环境变量?
May 13 Python
python实现图像检索的三种(直方图/OpenCV/哈希法)
Aug 08 Python
pytorch实现特殊的Module--Sqeuential三种写法
Jan 15 Python
详解Python中pyautogui库的最全使用方法
Apr 01 Python
python修改微信和支付宝步数的示例代码
Oct 12 Python
Python import模块的缓存问题解决方案
Jun 02 Python
谈一谈数组拼接tf.concat()和np.concatenate()的区别
Feb 07 #Python
python文件和文件夹复制函数
Feb 07 #Python
tf.concat中axis的含义与使用详解
Feb 07 #Python
浅谈tensorflow 中tf.concat()的使用
Feb 07 #Python
Python for循环通过序列索引迭代过程解析
Feb 07 #Python
python中with用法讲解
Feb 07 #Python
如何基于Python实现数字类型转换
Feb 07 #Python
You might like
PHP删除数组中特定元素的两种方法
2013/07/02 PHP
PHP使用CURL_MULTI实现多线程采集的例子
2014/07/29 PHP
jquery简单体验
2007/01/10 Javascript
javascript,jquery闭包概念分析
2010/06/19 Javascript
jQuery实现隔行背景色变色
2014/11/24 Javascript
NodeJS Web应用监听sock文件实例
2015/02/18 NodeJs
JavaScript 最佳实践:帮你提升代码质量
2016/12/03 Javascript
Chrome浏览器的alert弹窗禁止再次弹出后恢复的方法
2016/12/30 Javascript
layui分页效果实现代码
2017/05/19 Javascript
JavaScript模拟文件拖选框样式v1.0的实例
2017/08/04 Javascript
基于JavaScript实现简单的音频播放功能
2018/01/07 Javascript
mac上配置Android环境变量的方法
2018/07/08 Javascript
关于vue v-for 循环问题(一行显示四个,每一行的最右边那个计算属性)
2018/09/04 Javascript
实例介绍JavaScript中多种组合继承
2019/01/20 Javascript
关于vue状态过渡transition不起作用的原因解决
2019/04/09 Javascript
Python中元组,列表,字典的区别
2017/05/21 Python
tensor和numpy的互相转换的实现示例
2019/08/02 Python
python中的反斜杠问题深入讲解
2019/08/12 Python
Flask中endpoint的理解(小结)
2019/12/11 Python
如何通过python实现全排列
2020/02/11 Python
Django REST framwork的权限验证实例
2020/04/02 Python
加拿大领先的优质厨具产品在线购物网站:Golda’s Kitchen
2017/11/17 全球购物
意大利奢侈品多品牌集合店:TheDoubleF
2019/08/24 全球购物
俄罗斯药房连锁店:ASNA
2020/06/20 全球购物
在C中是否有模拟继承等面向对象程序设计特性的好方法
2012/05/22 面试题
经典c++面试题六
2012/01/18 面试题
一套比较完整的软件测试人员面试题
2012/05/13 面试题
新媒传信软件测试面试题
2013/02/24 面试题
Python面试题:如何用Python来发送邮件
2016/03/15 面试题
单位成立周年感言
2014/01/26 职场文书
公司面试感谢信
2014/02/01 职场文书
体育系毕业生自荐信
2014/06/28 职场文书
党员查摆剖析材料
2014/10/10 职场文书
php引用传递
2021/04/01 PHP
elasticSearch-api的具体操作步骤讲解
2021/06/28 Java/Android
Golang 并发下的问题定位及解决方案
2022/03/16 Golang