【Ubuntu-Tensorflow】InvalidArgumentError GPU不能使用的问题

问题内容如下

InvalidArgumentError (see above for traceback): Cannot assign a device to node 'train/final/fc3/b/Momentum': Could not satisfy explicit device specification '/device:GPU:0' because no devices matching that specification are registered in this process; available devices: /job:localhost/replica:0/task:0/cpu:0
Colocation Debug Info:
Colocation group had the following types and devices:
ApplyMomentum: CPU
Mul: CPU
Sum: CPU
Abs: CPU
Const: CPU
Assign: CPU
Identity: CPU
VariableV2: CPU [[Node: train/final/fc3/b/Momentum = VariableV2[_class=["loc:@train/final/fc3/b"], container="", dtype=DT_FLOAT, shape=[29], shared_name="", _device="/device:GPU:0"]()]]

问题原因：有些tf.Variable()不允许在GPU运行，需要在CPU运行
解决办法：修改配置里面的allow_soft_placement=True，而allow_soft_placement=True，就是运行在出现不允许在GPU运行的时候，可以切换到CPU运行。

【2017/8/9】20：20
经过一个下午的不断的调试认证，发现，allow_soft_placement=True，就是在GPU跑不动程序的时候，自动切换到CPU运行，这种方式虽然能够确保程序的正常运行，但是却不能保证GPU出现问题能够及时的反馈，故而还是推荐使用allow_soft_placement=True的选项。

【Ubuntu-Tensorflow】InvalidArgumentError GPU不能使用的问题相关推荐

Ubuntu tensorflow自定义GPU版本op节点
参考:https://blog.csdn.net/qq_27637315/article/details/79114633 windows增加op节点: https://github.com/tens ...
神经网络环境python2.7+tensorflow（gpu）+keras+cuda toolkit8.0+cudnn5.1+anaconda（ubuntu 16.04desktop64位）
tensorflow的GPU开发环境搭建有各种雷区,不是特别好布置,但是tensorflow开发环境又是目前运用很广泛的神经网络开发环境,所以我分享一下我的个人安装的经验. 首先是ubuntu的安装我 ...
没GPU也能玩梵高作画：Ubuntu tensorflow CPU版
没GPU也能玩梵高作画:Ubuntu tensorflow CPU版作者:七月在线开发/市场团队骁哲.李伟.July 时间:二零一六年九月二十七日交流:TensorFlow实战 ...
Win10 TensorFlow（gpu）安装详解
Win10 TensorFlow(gpu)安装详解写在前面:TensorFlow是谷歌基于DistBelief进行研发的第二代人工智能学习系统,其命名来源于本身的运行原理.Tensor(张量)意味着 ...
gpu跑普通python程序_普通电脑PC怎样跑TensorFlow的GPU模式
展开全部首先需要看你的PC配置是否32313133353236313431303231363533e4b893e5b19e31333363373736够,TF的GPU模式只支持N卡,然后计算能力高于 ...
tensorflow的GPU加速计算
参考 tensorflow的GPU加速计算 - 云+社区 - 腾讯云一.概述 tensorflow程序可以通过tf.device函数来指定运行每一个操作的设备,这个设备可以是本地的CPU或者G ...
2020-08-21 Windows 10安装 Anaconda+tensorflow 配置GPU
Windows 10安装 Anaconda+tensorflow 配置GPU 参考链接 https://blog.csdn.net/hitzijiyingcai/article/details/833 ...
Ubuntu Tensorflow object_detection API 目标检测环境搭建
Ubuntu 16.04下安装TensorFlow Object Detection API(对象检测API) Ubuntu 16.04下搭建TensorFlow运行环境(用Anaconda) ...
使用TensorFlow，GPU和Docker容器进行深度学习
摘要: 数据科学家使用GPU来提高TensorFlow的计算速度,但GPU价格昂贵,也需要对其所占用的资源进行认真的管理.本文将带你来一起解决这一问题. 在过去的几个月中,我和多个企业的数据科学团队进 ...
tensorflow with gpu 环境配置
1.准备工作 1.1 确保GPU驱动已经安装 lspci | grep -i nvidia 通过此命令可以查看GPU信息,测试机已经安装GPU驱动 nvidia-smi 可以查看英伟达显卡信息 1.2 ...

【Ubuntu-Tensorflow】InvalidArgumentError GPU不能使用的问题

【Ubuntu-Tensorflow】InvalidArgumentError GPU不能使用的问题相关推荐

最新文章

热门文章