Need help in setting up TensorRt

[restive cobalt]

i am using tensorflow gpu in wsl for training deep learning models in VScode . can someone please help me in setting up TensorRt step by step for training the models faster

[inland osprey]

The steps are at https://www.tensorflow.org/install/pip#step-by-step_instructions, just ignore the cuda toolkit, sdk, and tensorrt bit

You only need gpu drivers, WSL, and vs code

[inland osprey]

If you still haev trouble, post the error message and screenshots here

[restive cobalt]

can you tell me which version of tensorrt should i download?

[inland osprey]

you don't

oh I see, you already have gpu setup and are looking to add TensorRT to it

check out the python notebooks at https://github.com/tensorflow/tensorrt/tree/master/tftrt/benchmarking-python/image_classification

Is TensorRT a hard requirement for you? I'd only put the effort in setting it up if absolutely necessary, otherwise the gpu is plenty fast

[restive cobalt]

oh i see .. right now its not a hard requirement

i am going to avoid for now in that case😅

thank you for your time