Tensorflow: no supported kernel for GPU devices is available

最新推荐文章于 2023-10-23 09:55:53 发布

原创最新推荐文章于 2023-10-23 09:55:53 发布 · 1.5k 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#tensorflow #bug

python 同时被 2 个专栏收录

7 篇文章

订阅专栏

tensorflow

2 篇文章

订阅专栏

本文讨论了在使用TensorFlow时遇到的GPU兼容性问题，特别是在运行tf.gather和tf.image.resize_nearest_neighbor等操作时。由于缺乏对特定数据类型的CUDA内核支持，作者提出了一种解决策略，即通过数据类型转换来规避这一限制。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

几个tensorflow的错误，发现没有办法在GPU上跑
包括tf.gather, tf.image.resize_nearest_neighbor等，报如下错误：

ResizeNearestNeighbor: Could not satisfy explicit device specification '/device:GPU:0' because no supported kernel for GPU devices is available.

之前只能在前面加上with tf.device("/cpu:0"):
今天查了一下，是tf没有在CUDA运算中为这几种操作添加对应Data type的Registered kernels，解决方法可以手动添加。
目前我是通过hack的手段解决的，先把输入的tensor 做tf.cast到支持的数据类型(比如 float32,或int64, 视具体情况而定)，然后输入相应函数，输出后再cast回原数据类型

END