[苹果M1]：我没有注册与节点{{RngReadAndSkip RngReadAndSkip}}兼容的'GPU‘设备的’OpKernel‘节点。已注册:设备=‘CPU’
[苹果M1]：我没有注册与节点{{RngReadAndSkip RngReadAndSkip}}兼容的'GPU‘设备的’OpKernel‘节点。已注册:设备=‘CPU’
提问于 2021-09-07 12:52:44
这是我的代码
random = tf.random.Generator.from_seed(42)
random = random.normal(shape=(2,2))但是我得到了这个错误：
tensorflow.python.framework.errors_impl.NotFoundError: No registered 'RngReadAndSkip'
OpKernel for 'GPU' devices compatible with node {{node RngReadAndSkip}}. Registered: device='CPU'回答 1
Stack Overflow用户
发布于 2021-10-28 00:12:04
这是因为一些TF/Keras高级方法没有在tensorflow-metal中实现。您可以尝试此解决方法：
with tf.device('/cpu:0'):
... # your code which need RngReadAndSkip low-level method here接下来，你可以正常地做其他事情(使用GPU)。
希望，苹果有一天会修复它(还有他们的另一个成堆的bug，lmao)
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/69088577
复制相关文章
相似问题