当前位置:   article > 正文

tensorflow运行在gpu还是cpu_tensorflowgpu版可以用cpu吗

tensorflowgpu版可以用cpu吗

tensorflow在电脑的gpu和cpu上均可运行:

  • "/cpu:0":机器的 CPU。
  • "/device:GPU:0":机器的 GPU(如果有一个)。
  • "/device:GPU:1":机器的第二个 GPU(以此类推)。

当想要知道指令和张量在哪个设备上运行时,可以这样设置session

sess = tf.Session(config=tf.ConfigProto(log_device_placemnet=True))

手动分配设备

  1. with tf.device('/cpu:0'):
  2. ...
  3. sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

当使用GPU时,需要设置允许GPU增加内存,有两种设置方式:

(1)设置allow_grouth,根据运行时的需要来分配GPU内存

  1. config = tf.ConfigProto()
  2. config.gpu_options.allow_growth = True
  3. session = tf.Session(config=config,...)

(2)设置per_

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/小小林熬夜学编程/article/detail/181510
推荐阅读
相关标签
  

闽ICP备14008679号