Post marked as solved
Click to stop watching this thread.
You have stopped watching this post. Click to start watching again.
Post marked as solved with 2 replies, 0 views
Many thanks for your reply.
Currently I am using standard tensorflow (without the metal plugin & running on CPU) without any issues. I haven't tried your workaround, but I think you're right that it's the GPU calculations going wrong, so it makes sense using tf.device('CPU: 0'): would work too. So at least we have two workarounds for anyone else who encounters this issue.
Thank you for your help!