File "/root/anaconda3/envs/dirt/lib/python3.7/site-packages/tensorflow_core/python/framework/ops.py", line 6606, in raise_from_not_ok_status
six.raise_from(core._status_to_exception(e.code, message), None)
File "", line 3, in raise_from
tensorflow.python.framework.errors_impl.NotFoundError: Could not find valid device for node.
Node:{{node Rasterise}}
All kernels registered for op Rasterise :
device='GPU'
[Op:Rasterise] name: Rasterise/RasteriseBatch/
File "/root/anaconda3/envs/dirt/lib/python3.7/site-packages/tensorflow_core/python/framework/ops.py", line 6606, in raise_from_not_ok_status six.raise_from(core._status_to_exception(e.code, message), None) File "", line 3, in raise_from
tensorflow.python.framework.errors_impl.NotFoundError: Could not find valid device for node.
Node:{{node Rasterise}}
All kernels registered for op Rasterise :
device='GPU'
[Op:Rasterise] name: Rasterise/RasteriseBatch/