pycuda._driver.LogicError: cuMemcpyHtoDAsync failed: invalid argument

[lxl910915xh4ob]

The tensorrt engine input type is numpt.float32, but we give it an int array. In our code, we changed np.ones((1,1024, 1024, 3)) to np.ones((1, 1024, 1024, 3)).astype(np.float32)