Hi,
I think error could be because of cuda. As same program without CUDA Libtorch runs perfectly well(left terminal), whereas with CUDA Libtorch throws error(right terminal). Any help would be welcome.

image.png1299×736 189 KB

Thanks.