tensorflow.python.framework.errors_impl.NotFoundError #260

Traceback (most recent call last):
File "C:\Users\lzc\Desktop\SRGAN-master\train.py", line 120, in
G.init_build(tlx.nn.Input(shape=(8, 3, 96, 96)))
File "D:\anaconda3\Lib\site-packages\tensorlayerx\nn\core\core_tensorflow.py", line 634, in init_build
self.forward(*inputs, **kwargs)
File "C:\Users\lzc\Desktop\SRGAN-master\srgan.py", line 71, in forward
x = self.subpiexlconv1(x)
^^^^^^^^^^^^^^^^^^^^^
File "D:\anaconda3\Lib\site-packages\tensorlayerx\nn\core\core_tensorflow.py", line 173, in call
output = self.forward(inputs, *args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "D:\anaconda3\Lib\site-packages\tensorlayerx\nn\layers\convolution\super_resolution.py", line 184, in forward
outputs = self.depth_to_space(inputs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "D:\anaconda3\Lib\site-packages\tensorlayerx\backend\ops\tensorflow_backend.py", line 1675, in call
return tf.nn.depth_to_space(input, block_size=self.block_size, data_format=self.data_format)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "D:\anaconda3\Lib\site-packages\tensorflow\python\ops\weak_tensor_ops.py", line 88, in wrapper
return op(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^
File "D:\anaconda3\Lib\site-packages\tensorflow\python\util\traceback_utils.py", line 153, in error_handler
raise e.with_traceback(filtered_tb) from None
File "D:\anaconda3\Lib\site-packages\tensorflow\python\framework\ops.py", line 5983, in raise_from_not_ok_status
raise core._status_to_exception(e) from None # pylint: disable=protected-access
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^

tensorflow.python.framework.errors_impl.NotFoundError: Could not find device for node: {{node DepthToSpace}} = DepthToSpace[T=DT_FLOAT, block_size=2, data_format="NCHW"]
All kernels registered for op DepthToSpace:
device='CPU'; T in [DT_UINT64]; data_format in ["NHWC"]
device='CPU'; T in [DT_INT64]; data_format in ["NHWC"]
device='CPU'; T in [DT_UINT32]; data_format in ["NHWC"]
device='CPU'; T in [DT_UINT16]; data_format in ["NHWC"]
device='CPU'; T in [DT_INT16]; data_format in ["NHWC"]
device='CPU'; T in [DT_UINT8]; data_format in ["NHWC"]
device='CPU'; T in [DT_INT8]; data_format in ["NHWC"]
device='CPU'; T in [DT_INT32]; data_format in ["NHWC"]
device='CPU'; T in [DT_HALF]; data_format in ["NHWC"]
device='CPU'; T in [DT_BFLOAT16]; data_format in ["NHWC"]
device='CPU'; T in [DT_FLOAT]; data_format in ["NHWC"]
device='CPU'; T in [DT_DOUBLE]; data_format in ["NHWC"]
device='CPU'; T in [DT_COMPLEX64]; data_format in ["NHWC"]
device='CPU'; T in [DT_COMPLEX128]; data_format in ["NHWC"]
device='CPU'; T in [DT_BOOL]; data_format in ["NHWC"]
device='CPU'; T in [DT_STRING]; data_format in ["NHWC"]
device='CPU'; T in [DT_RESOURCE]; data_format in ["NHWC"]
device='CPU'; T in [DT_VARIANT]; data_format in ["NHWC"]
device='XLA_CPU_JIT'; T in [DT_FLOAT, DT_DOUBLE, DT_INT32, DT_UINT8, DT_INT16, DT_INT8, DT_COMPLEX64, DT_INT64, DT_BOOL, DT_QINT8, DT_QUINT8, DT_QINT32, DT_BFLOAT16, DT_UINT16, DT_COMPLEX128, DT_HALF, DT_UINT32, DT_UINT64, DT_FLOAT8_E5M2, DT_FLOAT8_E4M3FN, DT_INT4, DT_UINT4]