Tensorflow加载模型报错InvalidArgumentError (see above for traceback):Connot assign a device...

最新推荐文章于 2022-07-19 15:39:38 发布
最新推荐文章于 2022-07-19 15:39:38 发布
阅读量1k 收藏
点赞数
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/weixin_42228294/article/details/118069119
版权

错误提示：

InvalidArgumentError (see above for traceback): Cannot assign a device for operation G_paper_1/Run/G_paper_1/latents_in: node G_paper_1/Run/G_paper_1/latents_in (defined at /home/abc/Downloads/iGAN-HD-master/iGAN_HD-v2-1024/tfutil.py:508) was explicitly assigned to /device:GPU:0 but available devices are [ /job:localhost/replica:0/task:0/device:CPU:0, /job:localhost/replica:0/task:0/device:XLA_CPU:0 ]. Make sure the device specification refers to a valid device. The requested device appears to be a GPU, but CUDA is not enabled.

原因是tf里的一个bug，解决方法是设置配置里面的allow_soft_placement=True，当运行在出现不允许在GPU运行的时候，可以切换到CPU运行。

解决方法：
tf.ConfigProto()内添加allow_soft_placement=True

run_config = tf.ConfigProto(allow_soft_placement=True)
Mr.小蔡
关注
  • 0
    点赞
  • 踩
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
Ubuntu Yolov3 训练新的模型后，加载新模型参数出错InvalidArgumentError: Assign requires shapes of both tensors to match
wukai0909的博客
08-06 1109
--class_name_path ： 类名文件 --restore_path ：加载训练完成的模型 把这两个文件替换成自己训练的文件夹 tensorflow.python.framework.errors_impl.InvalidArgumentError: Assign requires shapes of both tensors to match. lhs shape= [255] ...
Tensorflow加载模型报错 Cannot assign a device for operation...
sjwzdh的博客
11-28 2863
报错 InvalidArgumentError (see above for traceback): Cannot assign a device for operation ‘tower_3_1/eval_outputs_and_metrics/foldl/while/generator/generator/rnn/while/rnn/dna_cell/assert_equal/Assert/...
tensorflow错误：InvalidArgumentError (see above for traceback): Cannot assign a device for operation
简简单单
01-02 7697
一、错误 InvalidArgumentError (see above for traceback): Cannot assign a device for operation 'shuffle_batch': Could not satisfy explicit device specification '/device:GPU:0' because no supported kernel ...
【tf.keras】：版本从1.x升级到2.x遇到的一个问题：InvalidArgumentError: Cannot assign a device for operation embedding_
Jack_Kuo的博客
07-19 652
代码】【tf.keras】版本从1.x升级到2.x遇到的一个问题InvalidArgumentErrorCannotassignadeviceforoperationembedding_
Tensorflow报错：Invalid argument: Cannot parse tensor from tensor_proto.
梁瑛平的博客
12-18 1121
这是由于网络复杂度过高，显存不足而导致的，不过一般网络不会占这么大内存，最好重新检查一下网络中特征层的大小变化情况。
解决tensorflow1.x版本加载saver.restore目录报错的问题
12-24
NotFoundError (see above for traceback): Unsuccessful TensorSliceReader constructor: Failed to find any matching files for xxx Traceback (most recent call last): File "C:\Users\Administrator\...
解决tensorflow测试模型时NotFoundError错误的问题
01-20
NotFoundError (see above for traceback): Unsuccessful TensorSliceReader constructor: Failed to find any matching files for xxx …… 经查资料分析，错误原因可能出在加载模型时的路径问题。我采用的加载...
【解决方案】pyspark 初次连接mongo 时报错Class not found exception:com.mongodb.spark.sql.DefaultSource
01-20
部分报错如下： Traceback (most recent call last):  File /home/cisco/spark-mongo-test.py, line 7, in  df = spark.read.format(com.mongodb.spark.sql.DefaultSource).load()  File /home/cisco/spark-...
问题解决：ImportError: No module named paddle.v2
01-20
Traceback (most recent call last): File D:/xxx.py, line 13, in from paddle.v2.plot import Ploter ModuleNotFoundError: No module named 'paddle.v2' 解决思路 更新到python3之后，这个包里的画图工具改变...
Python程序基础：Python中的异常.pptx
06-06
>>>a = 1 >>>if a == 1 print("Yes") File "<stdin>", line 1 if a == 1 print("Yes") ^ SyntaxError: invalid syntax 2．异常 程序在运行时也有可能发生错误，这种在运行期间检测到的错误称为异常。 例如： >>>'2' ...
用deeplab训练自己数据时出现 InvalidArgumentError (see above for traceback): [`labels` out of boundary]
Winters____的博客
12-11 1488
首先，明确问题问题错误点，是label值out of range,在制作标签时，难免会resize 原图和label，而问题就出现在resize的时候，我们在利用cv2,resize()时要注意方式选择 最近邻插值 方式，即 cv2.resize(src,(h,w),interpolation=cv2.INTER_NEAREST) 这个插值方式是根据最近像素值得值进行插值的，比如原label...
Cannot assign a device for operation ‘Gs/_Run/Gs/latents_in‘:Operation was explicitly assigned to GP
lanmy_dl的博客
03-14 2430
今天要迁移项目，本来我一直在本地跑的用anaconda3.6主里面，原来创建虚拟的环境方便迁移，这下我只有重新创建新环境重下包进行迁移，但是包都下好了却出现这样奇怪的错误，百度了说是找不到GPU，可是我之前都能跑啊换个环境没理由GPU没有跑不了？ 按照有些帖子在session那里写成这样： config = tf.ConfigProto(allow_soft_placement = True) sess = tf.Session(config = config) 又报错：Conv2DCustomBack
InvalidArgumentError (see above for traceback): Cannot parse tensor from proto: dtype: DT_FLOAT
slq1023的博客
09-27 2888
今天运行一个深度学习任务竟然报了这么长的错。 查了一些资料还没有解决，难道是内存太小？ 运行的同时打开任务管理器发现内存并没有达到100%就报错了。 来一位大神帮帮我吧。     python D:\TIANCHI\code\al1.py Using TensorFlow backend. (x_train.shape: (144, 1920, 2560) (y_train.shape...
Tensorflow 模型恢复时，设备指定与当前不匹配报错的解决方法
我的博客
05-17 1659
错误内容：nvalidArgumentError (see above for traceback): Cannot assign a device for operation ‘Reshape’: Operation was explicitly assigned to /job:worker/task:0 but available devices are [ /job:localhost/r...
tensorflow 在cpu的环境中无法导入gpu训练好的模型（Make sure the device specification refers to a valid device.）
weixin_39099017的博客
04-08 5039
在我准备迁移测试mobilenet时在调用new_saver.restore(sess, checkpoint_path_data)出现InvalidArgumentError: Cannot assign a device for operation 'clone_1/gradients/clone_1/MobileNet/conv_1/batch_norm/moments/sufficient...
Tensorflow错误汇总与解决办法
weixin_33943347的博客
12-19 2756
2017-12-19 盘点遇到的各种Tensorflow坑（此博客不定期更新）： １．InvalidArgumentError (see above for traceback): Cannot assign a device for operation 'shuffle_batch': Could not satisfy explicit device specification...
InvalidArgumentError (see above for traceback): Assign requires shapes of both tensors to match
摆渡者
07-31 6140
在服务器上训练完的模型，copy到本地电脑进行测试模型，但是此时的测试语料是新的语料，存在“测试语料中的字不在之前的训练时生成的字典中”（本人做的是基于LSTM的命名实体识别的任务）的问题，在测试的过程中，运动下面的问题： InvalidArgumentError (see above for traceback): Assign requires shapes of both tensors ...
[Bug集合]InvalidArgumentError (see above for traceback): targets[0] is out of range
Eminbogen的博客
11-08 2048
常见入门问题，在拷贝修改别人程序时一定概率出现。 原因：你一定是改了　神经网络输出层神经元个数　或　给予神经网络的训练集的标签数目　其中的一种，使得两者不匹配。 比如我拷贝了一个猫狗大战的程序改，结果数据集分了10类，但神经网络层输出神经元还是两个（原猫和狗），结果不匹配了。 如： def inference(images, batch_size, n_classes,keep_prob): ...
tensorflow-gpu遇到的问题
明夏小斯的博客
12-07 2871
遇到的第一个问题：InvalidArgumentError
InvalidArgumentError (see above for traceback): Cannot assign a device for operation 'IteratorToStringHandle': Operation was explicitly assigned to /device:GPU:0 but available devices are [ /job:localhost/replica:0/task:0/device:CPU:0 ]. Make sure the device specification refers to a valid device. [[{{node IteratorToStringHandle}} = IteratorToStringHandle[_device="/device:GPU:0"](IteratorV2)]]
最新发布
06-06
这个错误是因为您尝试在 GPU 上执行一个不支持 GPU 的操作。具体地说，在您的代码中，`IteratorToStringHandle` 这个操作被指定在了 `/device:GPU:0` 上执行，但是这个操作只能在 CPU 上运行。因此，TensorFlow 抛出了这个错误。 要解决这个问题，您需要将 `IteratorToStringHandle` 这个操作从 GPU 设备上移除，使其能够在 CPU 上运行。一种简单的方法是将 `with tf.device("/gpu:0"):` 改为 `with tf.device("/cpu:0"):`，这样包括 `IteratorToStringHandle` 在内的所有操作都将在 CPU 上运行。修改后的代码如下所示： ```python with tf.device("/cpu:0"): # 在这里添加您的 TensorFlow 训练代码 ``` 如果您需要将某些操作固定在 GPU 上执行，可以使用 `tf.device` 和 `tf.compat.v1.disable_eager_execution()` 来手动控制 TensorFlow 的计算图。具体来说，您可以在 `with tf.device("/gpu:0"):` 块中创建一个 TensorFlow 计算图，并使用 `tf.compat.v1.Session` 来运行这个计算图。这样，您就可以将某些操作固定在 GPU 上运行，而将其他操作放在 CPU 上运行。下面是一个示例代码： ```python import tensorflow as tf # 关闭 Eager Execution tf.compat.v1.disable_eager_execution() # 定义一个 TensorFlow 计算图 with tf.device("/gpu:0"): x = tf.constant([1.0, 2.0, 3.0]) y = tf.constant([4.0, 5.0, 6.0]) z = tf.add(x, y) # 创建一个 TensorFlow 会话并运行计算图 with tf.compat.v1.Session() as sess: # 将 z 固定在 GPU 上运行 result = sess.run(z) print(result) ``` 在这个示例代码中，`x` 和 `y` 这两个操作被固定在 GPU 上执行，而 `z` 这个操作则会被自动放到 GPU 上执行。最终，您将获得一个包含 `[5.0, 7.0, 9.0]` 的 NumPy 数组。

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值