1200字范文,内容丰富有趣,写作的好帮手!
1200字范文 > TensorRT报Cuda initialization failure with error

TensorRT报Cuda initialization failure with error

时间:2023-10-30 11:07:20

相关推荐

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: /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问题就解决了

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。