I am running TensorRT inference using a model written in tensorflow and converted into uff. The model is saved with input size of (1, 601, 400). However, when I try to create tensorRT engine (at calling buildCudaEngine function) with input height set to 480, I get the following error:
INFO: --------------- Timing deconv4/conv2d_transpose(4) INFO: Tactic 1 time 2.20518 ERROR: c:\p4sw\sw\gpgpu\MachineLearning\DIT\release.0\builder\cudnnBuilderUtils.cpp (255) - Cuda Error in nvinfer1::cudnn::findFastestTactic: 77 ERROR: c:\p4sw\sw\gpgpu\MachineLearning\DIT\release.0\engine\runtime.cpp (30) - Cuda Error in nvinfer1::`anonymous-namespace'::DefaultAllocator::free: 77
If I change input height to 478 or 481, not changing anything else at all, engine builds and I can run inference without any issues. Height of 479 does not work. I tried creating engine using a different uff file with input set to (1, 480, 400), still the same error. Why could I possibly be getting this error and how can I fix it?