wap手机网站建设方案,时间线 wordpress,怎么建网站站点,wordpress 评论框插件1.指定GPU运算如果安装的是GPU版本#xff0c;在运行的过程中TensorFlow能够自动检测。如果检测到GPU#xff0c;TensorFlow会尽可能的利用找到的第一个GPU来执行操作。如果机器上有超过一个可用的GPU#xff0c;除了第一个之外的其他的GPU默认是不参与计算的。为了让Tensor…1.指定GPU运算如果安装的是GPU版本在运行的过程中TensorFlow能够自动检测。如果检测到GPUTensorFlow会尽可能的利用找到的第一个GPU来执行操作。如果机器上有超过一个可用的GPU除了第一个之外的其他的GPU默认是不参与计算的。为了让TensorFlow使用这些GPU必须将OP明确指派给他们执行。with......device语句能够用来指派特定的CPU或者GPU执行操作import tensorflow as tfimport numpy as npwith tf.Session() as sess:with tf.device(/cpu:0):a tf.placeholder(tf.int32)b tf.placeholder(tf.int32)add tf.add(a, b)sum sess.run(add, feed_dict{a: 3, b: 4})print(sum)设备的字符串标识当前支持的设备包括以下的几种cpu:0 机器的第一个cpu。gpu:0 机器的第一个gpu如果有的话gpu:1 机器的第二个gpu依次类推类似的还有tf.ConfigProto来构建一个config在config中指定相关的GPU并且在session中传入参数config“自己创建的config”来指定gpu操作其中tf.ConfigProto函数的参数如下log_device_placementTrue: 是否打印设备分配日志allow_soft_placementTrue: 如果指定的设备不存在允许TF自动分配设备import tensorflow as tfimport numpy as npconfig tf.ConfigProto(log_device_placementTrue, allow_soft_placementTrue)with tf.Session(configconfig) as sess:a tf.placeholder(tf.int32)b tf.placeholder(tf.int32)add tf.add(a, b)sum sess.run(add, feed_dict{a: 3, b: 4})print(sum)2.设置GPU使用资源上文的tf.ConfigProto函数生成的config之后还可以设置其属性来分配GPU的运算资源如下代码就是按需分配import tensorflow as tfimport numpy as npconfig tf.ConfigProto(log_device_placementTrue, allow_soft_placementTrue)config.gpu_options.allow_growth Truewith tf.Session(configconfig) as sess:a tf.placeholder(tf.int32)b tf.placeholder(tf.int32)add tf.add(a, b)sum sess.run(add, feed_dict{a: 3, b: 4})print(sum)使用allow_growth option刚开始会分配少量的GPU容量然后按需要慢慢的增加有与不会释放内存随意会导致内存碎片。同样上述的代码也可以在config创建时指定import tensorflow as tfimport numpy as npgpu_options tf.GPUOptions(allow_growthTrue)config tf.ConfigProto(gpu_optionsgpu_options)with tf.Session(configconfig) as sess:a tf.placeholder(tf.int32)b tf.placeholder(tf.int32)add tf.add(a, b)sum sess.run(add, feed_dict{a: 3, b: 4})print(sum)我们还可以给gpu分配固定大小的计算资源。gpu_options tf.GPUOptions(allow_growthTrue, per_process_gpu_memory_fraction0.5)上述代码的含义是分配给tensorflow的GPU显存大小为GPU的实际显存*0.5到此这篇关于tensorflow指定CPU与GPU运算的方法实现的文章就介绍到这了,更多相关tensorflow指定CPU与GPU运算内容请搜索我们以前的文章或继续浏览下面的相关文章希望大家以后多多支持我们本文标题: tensorflow指定CPU与GPU运算的方法实现本文地址: http://www.cppcns.com/jiaoben/python/309046.html
