TensorRT报Cuda initialization failure with error
Posted 修炼之路
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了TensorRT报Cuda initialization failure with error相关的知识,希望对你有一定的参考价值。
起因
在使用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.com/cuda/cuda-installation-guide-linux/index.html
问题原因
这种错误的原因是因为NVIDIA
驱动版本导致的,我用的是NVIDA官方的TensorRT
镜像nvcr.io/nvidia/tensorrt:21.03-py3
,在另外两台服务器上使用Driver Version: 460.84
和Driver Version: 460.80
都能够运行。
而出错这台服务器的NVIDIA
驱动的版本是460.56
解决办法
将NVIDIA
驱动的版本升级到Driver Version: 460.84
问题就解决了
以上是关于TensorRT报Cuda initialization failure with error的主要内容,如果未能解决你的问题,请参考以下文章
paddle报ValueError: (InvalidArgument) Pass tensorrt_subgraph_pass has not been registered
paddle报ValueError: (InvalidArgument) Pass tensorrt_subgraph_pass has not been registered
paddle报ValueError: (InvalidArgument) Pass tensorrt_subgraph_pass has not been registered