Closed lanyeeee closed 3 months ago
update TensorRT, and don't forget to modify the Environment Variables
After an afternoon of troubleshooting, I realized that I had forgotten to modify the environment variables for TensortRT, I was simply updating TensorRT via the .whl
file.
This results in TensortRT appearing to Python to have been updated, when in fact the TensortRT in use is still old (the one corresponds to the environment variable).
Now I modify the environment variable to the 8.6.1
version of TensortRT, and it works like a charm.
I tried to export TensorRT with this code
output:
Environment
TensorRT Version: 8.6.1 NVIDIA GPU: 3070Ti NVIDIA Driver Version: 522.06 CUDA Version: 11.8 CUDNN Version: 8.9.7.29 Operating System: Windows11 Python Version: 3.9 PyTorch Version: 2.3.1+cu118
In addition to above environment, I have also tried
CUDA-11.6 + CUDNN-8.7.0.84 + TensorRT-8.4.0
CUDA-11.6 + CUDNN-8.7.0.84 + TensorRT-8.5.1
CUDA-11.6 + CUDNN-8.7.0.84 + TensorRT-8.5.3
CUDA-11.6 + CUDNN-8.7.0.84 + TensorRT-8.6.0
CUDA-11.6 + CUDNN-8.7.0.84 + TensorRT-8.6.1
In all of these environments, the above error occurs.
I did find some similar issues #129 #75 but they did not helped.