在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中的yield浅析
Jun 16 Python
django 自定义用户user模型的三种方法
Nov 18 Python
Python 转义字符详细介绍
Mar 21 Python
python爬虫使用cookie登录详解
Dec 27 Python
python实现快速排序的示例(二分法思想)
Mar 12 Python
Python操作列表常用方法实例小结【创建、遍历、统计、切片等】
Oct 25 Python
Python箱型图处理离群点的例子
Dec 09 Python
tensorflow 初始化未初始化的变量实例
Feb 06 Python
pycharm 中mark directory as exclude的用法详解
Feb 14 Python
Python文件操作基础流程解析
Mar 19 Python
Python中内建模块collections如何使用
May 27 Python
教你怎么用python爬取爱奇艺热门电影
May 20 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
56.com视频采集接口程序(PHP)
2007/09/22 PHP
php 文件上传类代码
2011/08/06 PHP
微信公众号点击菜单即可打开并登录微站的实现方法
2014/11/14 PHP
PHP简单留言本功能实现代码
2017/06/09 PHP
Laravel5.1 框架Request请求操作常见用法实例分析
2020/01/04 PHP
js 新浪的一个图片播放图片轮换效果代码
2008/07/15 Javascript
用JS操作FRAME中的IFRAME及其内容的实现代码
2008/07/26 Javascript
multiSteps 基于Jquery的多步骤滑动切换插件
2011/07/22 Javascript
基于jQuery+HttpHandler实现图片裁剪效果代码(适用于论坛, SNS)
2011/09/02 Javascript
JS定时刷新页面及跳转页面的方法
2013/07/04 Javascript
告诉你什么是javascript的回调函数
2014/09/04 Javascript
jQuery validate插件submitHandler提交导致死循环解决方法
2016/01/21 Javascript
JS简单测试循环运行时间的方法
2016/09/04 Javascript
JavaScript判断变量名是否存在数组中的实例
2017/12/28 Javascript
分享ES6的7个实用技巧
2018/01/18 Javascript
vue中$refs, $emit, $on, $once, $off的使用详解
2019/05/26 Javascript
Vuex模块化应用实践示例
2020/02/03 Javascript
vue实现简单跑马灯效果
2020/05/25 Javascript
javascript使用正则表达式实现注册登入校验
2020/09/23 Javascript
[02:11]2016国际邀请赛中国区预选赛全程回顾
2016/07/01 DOTA
[02:32]DOTA2完美大师赛场馆静安体育中心观赛全攻略
2017/11/08 DOTA
[10:21]2018DOTA2国际邀请赛寻真——Winstrike
2018/08/11 DOTA
[01:03:27]Optic vs VGJ.S 2018国际邀请赛小组赛BO2 第一场 8.17
2018/08/20 DOTA
[01:47]2018年度DOTA2最佳教练-完美盛典
2018/12/16 DOTA
Python Socket编程详细介绍
2017/03/23 Python
Python实现中文数字转换为阿拉伯数字的方法示例
2017/05/26 Python
Tensorflow加载预训练模型和保存模型的实例
2018/07/27 Python
CSS3 制作绽放的莲花采用效果叠加实现
2013/01/31 HTML / CSS
深入浅析CSS3中的Flex布局整理
2020/04/27 HTML / CSS
保证书格式范文
2014/04/28 职场文书
工作分析计划书
2014/04/30 职场文书
北京奥运会口号
2014/06/21 职场文书
演讲稿:​快乐,从不抱怨开始!
2019/04/02 职场文书
小程序实现悬浮按钮的全过程记录
2021/10/16 HTML / CSS
解决Laravel使用验证时跳转到首页的问题
2021/11/17 PHP
使用 Koa + TS + ESLlint 搭建node服务器的过程详解
2022/05/30 NodeJs