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}}]]