admin 管理员组

文章数量: 1184232

起因

在使用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 问题就解决了

本文标签： 起因 驱动的版 台服务器

版权声明：本文标题：TensorRT报Cuda initialization failure with error_[trt] [w] cuda initialization failure with error: 内容由网友自发贡献，该文观点仅代表作者本人， 转载请联系作者并注明出处：http://www.roclinux.cn/p/1773818530a3566067.html， 本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容，一经查实，本站将立刻删除。