Getting error as Cuda Runtime (invalid argument)

Hi,

Please check the below topic for more info:

Thanks.