Cuda Error on running NGX SDK on AWS P3

While trying to run the demo code for image super resolution(NGX ISR) on AWS p3.2xlarge, I get the error saying error allocating output image CUDA buffer. Is this NGX not supported on P3?
Any help would be appreciated.

Additional info:
OS: Windows Server 2019
Using: Visual Studio Community 2019