起因

在使用TensorRT將onnx模型轉(zhuǎn)換為engine時(shí)報(bào)[TRT] Cuda initialization failure with error詳細(xì)錯(cuò)誤信息如下

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

問題原因

這種錯(cuò)誤的原因是因?yàn)镹VIDIA驅(qū)動(dòng)版本導(dǎo)致的，我用的是NVIDA官方的TensorRT鏡像nvcr.io/nvidia/tensorrt:21.03-py3，在另外兩臺(tái)服務(wù)器上使用Driver Version: 460.84和Driver Version: 460.80都能夠運(yùn)行。

而出錯(cuò)這臺(tái)服務(wù)器的NVIDIA驅(qū)動(dòng)的版本是460.56

解決辦法

將NVIDIA驅(qū)動(dòng)的版本升級(jí)到Driver Version: 460.84問題就解決了

總結(jié)

以上是生活随笔為你收集整理的TensorRT报Cuda initialization failure with error的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。