起因
在使用TensorRT将
onnx
模型转换为
engine
时报
[TRT] Cuda initialization failure with error
详细错误信息如下
[TensorRT] ERROR: CUDA initialization failure with error 222. Please check your CUDA installation:
问题原因
这种错误的原因是因为
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
问题就解决了


发布评论