TensorFlow学习系列(一)——GPU指定
来源:互联网 发布:怎么做淘宝客链接 编辑:程序博客网 时间:2024/05/18 02:03
1)一般不需要显示指定CPU还是GPU,TensorFlow能自动检测。
如果检测到GPU,TensorFlow会尽可能地利用找到的第一个GPU来执行操作。
如果机器上有超过一个可用的GPU,除第一个外的其他GPU默认是不参与计算的。为了让TensorFlow使用这些GPU,你必须将op明确指派给它们执行。with … Device语句用来指派特定的CPU或GPU执行操作:
with tf.Session() as sess: with tf.device("/gpu:1"): matrix1 = tf.constant([[3., 3.]]) matrix2 = tf.constant([[2.],[2.]]) product = tf.matmul(matrix1, matrix2)
设备用字符串进行标识. 目前支持的设备包括:"/cpu:0": 机器的 CPU."/gpu:0": 机器的第一个 GPU, 如果有的话."/gpu:1": 机器的第二个 GPU, 以此类推.
阅读全文
0 0
- TensorFlow学习系列(一)——GPU指定
- TensorFlow学习系列(一):初识TensorFlow
- tensorflow学习系列(一)
- tensorflow 指定GPU
- Tensorflow 指定GPU运行
- GPU编程系列之(一)——环境配置
- tensorflow学习(一)tensorflow-GPU安装(Driver375.26+CUDA8.0.61+CUDNN5.1)
- Tensorflow指定使用哪个GPU
- 指定TensorFlow使用哪一个GPU
- tensorflow指定gpu的使用
- window10 GPU版tensorflow安装(一)
- tensorflow学习笔记(一)——tensorflow基本使用
- 【tensorflow 学习】 gpu使用
- TensorFlow深度学习入门——Win10下安装Tensorflow和GPU加速
- 【深度学习】CentOS 7 安装GPU版Tensorflow教程(一)
- tensorflow学习笔记(二):windows10_64 tensorflow-gpu cuda
- (转)tensorflow中使用指定的GPU及GPU显存
- Tensorflow学习系列(1)——MNIST手写识别
- Spring入门到应用
- Java POI 写入Excel
- 面试题积累
- android开源框架RxPermissions动态权限申请
- linux命令之查看程序动态库依赖-ldd
- TensorFlow学习系列(一)——GPU指定
- Mybatis处理枚举类型
- openwrt 添加自定义内核模块
- 线程
- node mongoose
- js异步执行用回调函数当参数和同步执行的区别的案例
- 697 Degree of an Array
- 微信小程序服务器搭建
- 血压计代码学习笔记2017.6.2