Post

Replies

Boosts

Views

Activity

No registered 'RngReadAndSkip', tensor flow-metal 0.3.0
Hey, I got my new MBP M1 Max and started setting up the machine for tensorflow. I installed tensorflow-metal as instructed here: https://developer.apple.com/metal/tensorflow-plugin/ Here are some details: tensorflow-deps           2.7.0                         0    apple tensorflow-estimator      2.7.0                    pypi_0    pypi tensorflow-macos          2.7.0                    pypi_0    pypi tensorflow-metal          0.3.0                    pypi_0    pypi tensorflow-probability    0.7                        py_1    conda-forge When instantiating a keras subclassed model, I got the following error: NotFoundError: No registered 'RngReadAndSkip' OpKernel for 'GPU' devices compatible with node {{node RngReadAndSkip}} . Registered: device='CPU' [Op:RngReadAndSkip] The code works fine on my new machine when I uninstall tensorflow-metal. Looking forward to any probable solution. Rooting for running some TF projects on GPU!
4
0
2k
Jan ’23