TensorFlow指定CPU和GPU方法
TensorFlow指定CPU和GPU方法
TensorFlow 支持 CPU 和 GPU。它也支持分布式计算。可以在一个或多个计算机系统的多个设备上使用 TensorFlow。
TensorFlow 将支持的 CPU 设备命名为“/device:CPU:0”(或“/cpu:0”),第 i 个 GPU 设备命名为“/device:GPU:I”(或“/gpu:I”)。
如前所述,GPU 比 CPU 要快得多,因为它们有许多小的内核。然而,在所有类型的计算中都使用 GPU 也并不一定都有速度上的优势。有时,比起使用 GPU 并行计算在速度上的优势收益,使用 GPU 的其他代价相对更为昂贵。
为了解决这个问题,TensorFlow 可以选择将计算放在一个特定的设备上。默认情况下,如果 CPU 和 GPU 都存在,TensorFlow 会优先考虑 GPU。
TensorFlow 将设备表示为字符串。文展示如何在 TensorFlow 中指定某一设备用于矩阵乘法的计算。
具体做法
• 要验证 TensorFlow 是否确实在使用指定的设备(CPU 或 GPU),可以创建会话,并将 log_device_placement 标志设置为 True,即:
• 如果你不确定设备,并希望 TensorFlow 选择现有和受支持的设备,则可以将 allow_soft_placement 标志设置为 True:
• 手动选择 CPU 进行操作:
得到以下输出:
可以看到,在这种情况下,所有的设备都是 ‘/cpu:0’。
• 手动选择一个 GPU 来操作:
输出现在更改为以下内容:
每个操作之后的’/cpu:0’现在被替换为’/gpu:0’。
• 手动选择多个GPU:
在这种情况下,如果系统有 3 个 GPU 设备,那么第一组乘法将由’/:gpu:1’执行,第二组乘以’/gpu:2’执行。
解读分析
函数 tf.device() 选择设备(CPU 或 GPU)。with 块确保设备被选择并用于其操作。with 块中定义的所有变量、常量和操作将使用在 tf.device() 中选择的设备。
会话配置使用 tf.ConfigProto 进行控制。通过设置 allow_soft_placement 和 log_device_placement 标志,表示 TensorFlow 在指定的设备不可用时自动选择可用的设备,并在执行会话时给出日志消息作为描述设备分配的输出。
TensorFlow指定CPU和GPU方法相关推荐
- python调用gpu进行运算_tensorflow指定CPU与GPU运算的方法实现
1.指定GPU运算 如果安装的是GPU版本,在运行的过程中TensorFlow能够自动检测.如果检测到GPU,TensorFlow会尽可能的利用找到的第一个GPU来执行操作. 如果机器上有超过一个可用 ...
- tensorflow 制定 CPU 或GPU
tensorflow 制定 CPU 或GPU CPU with tf.Session() as sess:...: with tf.device("/cpu:0"):...: m1 ...
- tensorflow 指定cpu 但是还会运行 gpu_PyTorch VS TensorFlow 谁最强?这是标星 15000+ Transformers 库的运行结果...
点击上方"AI有道",选择"星标"公众号 重磅干货,第一时间送达 作者 | Lysandre Debut译者 | 陆离出品 | AI科技大本营(ID: rgzn ...
- tensorflow使用cpu不用gpu的问题
前提:cuda版本和cudnn版本已安装好,版本对应没问题: 问题:在安装好cuda.tensorflow.tensorflow-gpu的环境下,调用tensorflow跑模型,发现cpu资源占满,而 ...
- TensorFlow测试CPU、GPU
目录 1.查看当前Tensorflow版本 2.查看当前主机运行的设备 3.查看GPU是否可用,指定在CPU/GPU运行 4.比较CPU和GPU上的运行时间 1.查看当前Tensorflow版本 im ...
- Tensorflow 使用cpu和gpu的区别
在Tensorflow中使用gpu和cpu是有很大的差别的.在小数据集的情况下,cpu和gpu的性能差别不大.不过在大数据集的情况下,cpu的时间显著增加,而gpu变化并不明显.不过,我的笔记本电脑的 ...
- tensorflow的CPU和GPU的区别
最近深度学习CPU根本不够用,老板配置了GPU 怎么能不会用呢? 原来从来没有关注过的底层东西,于是乎痛下决心搞清楚原理和操作 先来看看第一部分 为什么GPU比CPU更diao呢? 这里就需要从他么的 ...
- 解决linux的-bash: ./xx: Permission denied/tensorflow 运行cpu还是gpu的方法
在linux下执行sh文件时提示下面信息: -bash: ./xx.sh: Permission denied 解决: chmod 777 xx.sh 今天在公司的服务器上用tensorflow训练, ...
- Tensorflow选择CPU或GPU运行
代码 import os os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" os.environ["CUD ...
最新文章
- python包管理器修改镜像地址
- 详解shuffle过程(转载)
- laravel项目运行 php artisan cache:clear 命令报错
- oracle 0 函数吗,Oracle9.2.0.1版函数编译一个小问题
- html转换pdf软件,html转换成高质量的PDF文件[跨平台开源软件]
- c语言5的阶乘流程图_5 种前途迷茫的编程语言
- python-描述符基本
- mysql筛选向导后返回原表_Navcat关于导出向导的常见问题!
- 如何获取查询生成器以字符串形式输出其原始SQL查询?
- WAF Bypass数据库特性(Oracle探索篇)
- 常用的分析方法论及分析框架
- 阿里云域名部署免费证书,开启https访问。
- U盘显示0字节怎么恢复完整解决教程
- 暗影骑士vn7安装linux,宏碁暗影骑士vn7 591g笔记本怎么样?
- MindMaster思维导图及亿图图示会员 超值获取途径
- Good Bye 2022: 2023 is NEAR 题解
- python中元组拆包_Python3基础:元组拆包
- 墨西哥付款方式有哪些
- TopResumedActivityChangeItem
- 2021最漂亮的5张可视化图
热门文章
- 从命令行到IDE,版本管理工具Git详解(远程仓库创建+命令行讲解+IDEA集成使用)
- Web项目使用nginx实现代理端口访问,看这篇就够了
- C/C++ 编程规范(01 )— 排版
- 2022-2028年中国阻燃纤维行业市场研究及前瞻分析报告
- Linux shell 学习笔记(2)— 监测程序、磁盘空间和处理文件(ps -ef、top、kill、df、du 、grep、tar)
- Python 标准库之 subprocesss
- CVD和ALD薄膜沉积技术应用领域
- 利用NVIDIA-NGC中的MATLAB容器加速语义分割
- NVIDIA Jarvis:一个GPU加速对话人工智能应用的框架
- [JAVA EE]ajax 方式提交数据