cuDNN error on tlt-evaluate ssd and tlt-infer ssd

Hi, I tried to run the ssd example jupyter notebook with RTX gpus, I successfully trained, pruned and retrain the model, but when I want to run tlt-evaluate or tlt-infer commands I got following error in jupyter:

tensorflow.python.framework.errors_impl.UnknownError: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above.
	 [[{{node model_1/conv1/convolution}}]]
	 [[{{node model_1/decoded_predictions/loop_over_batch/TensorArrayStack/TensorArrayGatherV3}}]]

Please try
$ export TF_FORCE_GPU_ALLOW_GROWTH=true

Reference: Unable to train SSD-Resnet-18

tnx