admin管理员组

文章数量:1618706

起因

在使用TensorRT将onnx模型转换为engine时报[TRT] Cuda initialization failure with error详细错误信息如下

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

问题原因

这种错误的原因是因为NVIDIA驱动版本导致的,我用的是NVIDA官方的TensorRT镜像nvcr.io/nvidia/tensorrt:21.03-py3,在另外两台服务器上使用Driver Version: 460.84Driver Version: 460.80都能够运行。

而出错这台服务器的NVIDIA驱动的版本是460.56

解决办法

NVIDIA驱动的版本升级到Driver Version: 460.84问题就解决了

本文标签: CUDATensorRTinitializationErrorfailure