TensorRT: Cuda initialization failure with error

created at 09-08-2021 views: 44

cause

When using TensorRT to convert the onnx model to engine, the [TRT] Cuda initialization failure with error. Detailed error information is as follows

[TensorRT] ERROR: CUDA initialization failure with error 222. Please check your CUDA installation: http://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html

problem causes

The reason for this error is due to the NVIDIA driver version. I used the official TensorRT mirror nvcr.io/nvidia/tensorrt:21.03-py3 of NVIDA. I used Driver Version: 460.84 and Driver Version: 460.80 on the other two servers. Can run.

And the NVIDIA driver version of this server is 460.56.

solution

Upgrade the NVIDIA driver version to Driver Version: 460.84 to solve the problem

created at:09-08-2021
edited at: 09-08-2021: