详解tf.device()指定tensorflow运行的GPU或CPU设备实现


Posted in Python onFebruary 20, 2021

在tensorflow中,我们可以使用 tf.device() 指定模型运行的具体设备,可以指定运行在GPU还是CUP上,以及哪块GPU上。

设置使用GPU

使用 tf.device('/gpu:1') 指定Session在第二块GPU上运行:

import tensorflow as tf
 
with tf.device('/gpu:1'):
  v1 = tf.constant([1.0, 2.0, 3.0], shape=[3], name='v1')
  v2 = tf.constant([1.0, 2.0, 3.0], shape=[3], name='v2')
  sumV12 = v1 + v2
 
  with tf.Session(config=tf.ConfigProto(log_device_placement=True)) as sess:
    print sess.run(sumV12)

ConfigProto() 中参数 log_device_placement=True  会打印出执行操作所用的设备,以上输出:

详解tf.device()指定tensorflow运行的GPU或CPU设备实现

如果安装的是GPU版本的tensorflow,机器上有支持的GPU,也正确安装了显卡驱动、CUDA和cuDNN,默认情况下,Session会在GPU上运行:

import tensorflow as tf
 
v1 = tf.constant([1.0, 2.0, 3.0], shape=[3], name='v1')
v2 = tf.constant([1.0, 2.0, 3.0], shape=[3], name='v2')
sumV12 = v1 + v2
 
with tf.Session(config=tf.ConfigProto(log_device_placement=True)) as sess:
  print sess.run(sumV12)

默认在GPU:0上执行:

详解tf.device()指定tensorflow运行的GPU或CPU设备实现

设置使用cpu

tensorflow中不同的GPU使用/gpu:0和/gpu:1区分,而CPU不区分设备号,统一使用 /cpu:0

import tensorflow as tf
 
with tf.device('/cpu:0'):
  v1 = tf.constant([1.0, 2.0, 3.0], shape=[3], name='v1')
  v2 = tf.constant([1.0, 2.0, 3.0], shape=[3], name='v2')
  sumV12 = v1 + v2
 
  with tf.Session(config=tf.ConfigProto(log_device_placement=True)) as sess:
    print sess.run(sumV12)

详解tf.device()指定tensorflow运行的GPU或CPU设备实现

到此这篇关于详解tf.device()指定tensorflow运行的GPU或CPU设备实现的文章就介绍到这了,更多相关tensorflow运行GPU或CPU内容请搜索三水点靠木以前的文章或继续浏览下面的相关文章希望大家以后多多支持三水点靠木!

Python 相关文章推荐
在Python的Flask框架下使用sqlalchemy库的简单教程
Apr 09 Python
解读Python编程中的命名空间与作用域
Oct 16 Python
numpy自动生成数组详解
Dec 15 Python
浅谈pyqt5中信号与槽的认识
Feb 17 Python
Python Numpy计算各类距离的方法
Jul 05 Python
python3实现斐波那契数列(4种方法)
Jul 15 Python
python 采用paramiko 远程执行命令及报错解决
Oct 21 Python
python、PyTorch图像读取与numpy转换实例
Jan 13 Python
Python实现Canny及Hough算法代码实例解析
Aug 06 Python
Python Pillow(PIL)库的用法详解
Sep 19 Python
Python3中FuzzyWuzzy库实例用法
Nov 18 Python
AI:如何训练机器学习的模型
Apr 16 Python
Python 的 f-string 可以连接字符串与数字的原因解析
Feb 20 #Python
安装不同版本的tensorflow与models方法实现
Feb 20 #Python
python爬虫scrapy基本使用超详细教程
Feb 20 #Python
解决pip安装tensorflow中出现的no module named tensorflow.python 问题方法
Feb 20 #Python
conda安装tensorflow和conda常用命令小结
Feb 20 #Python
TensorFlow低版本代码自动升级为1.0版本
Feb 20 #Python
python爬虫scrapy基于CrawlSpider类的全站数据爬取示例解析
Feb 20 #Python
You might like
PHP define函数的使用说明
2008/08/27 PHP
PHP 字符串长度判断效率更高的方法
2014/03/02 PHP
PHP中类的继承和用法实例分析
2016/05/24 PHP
PHP让数组中有相同值的组成新的数组实例
2017/12/31 PHP
PHP递归的三种常用方式
2019/02/28 PHP
PHP DB 数据库连接类定义与用法示例
2019/03/11 PHP
php中加密解密DES类的简单使用方法示例
2020/03/26 PHP
JavaScript判断两种格式的输入日期的正确性的代码
2007/03/25 Javascript
jquery BS,dialog控件自适应大小
2009/07/06 Javascript
jQuery模拟超链接点击效果代码
2013/04/21 Javascript
JS实现选择TextArea内文本的方法
2015/08/03 Javascript
Javascript 5种方法实现过滤删除前后所有空格
2016/06/22 Javascript
js 递归和定时器的实例解析
2017/02/03 Javascript
jQuery插件HighCharts绘制2D柱状图、折线图的组合双轴图效果示例【附demo源码下载】
2017/03/09 Javascript
详解Angular如何正确的操作DOM
2018/07/06 Javascript
bootstrap 路径导航 分页 进度条的实例代码
2018/08/06 Javascript
利用js将ajax获取到的后台数据动态加载至网页中的方法
2018/08/08 Javascript
React 项目迁移 Webpack Babel7的实现
2018/09/12 Javascript
d3绘制基本的柱形图的实现代码
2018/12/12 Javascript
vue中node_modules中第三方模块的修改使用详解
2019/05/31 Javascript
实例分析JS中的相等性判断===、 ==和Object.is()
2019/11/17 Javascript
[02:41]2015国际邀请赛中国区预选赛观战指南
2015/05/20 DOTA
python中的代码编码格式转换问题
2015/06/10 Python
python的文件操作方法汇总
2017/11/10 Python
numpy中矩阵合并的实例
2018/06/15 Python
对python多线程中互斥锁Threading.Lock的简单应用详解
2019/01/11 Python
详解如何从TensorFlow的mnist数据集导出手写体数字图片
2019/08/05 Python
python安装virtualenv虚拟环境步骤图文详解
2019/09/18 Python
python 解决函数返回return的问题
2020/12/05 Python
西班牙第一的网上药房:PromoFarma.com
2017/04/17 全球购物
aden + anais官方网站:婴儿襁褓、毯子、尿布和服装
2017/06/21 全球购物
机关门卫制度
2014/02/01 职场文书
公务员党的群众路线教育实践活动学习心得体会
2014/10/30 职场文书
MySQL的安装与配置详细教程
2021/06/26 MySQL
redis调用二维码时的不断刷新排查分析
2022/04/01 Redis
MySQL数据库之存储过程 procedure
2022/06/16 MySQL