TensorRT报Cuda initialization failure with error
生活随笔
收集整理的這篇文章主要介紹了
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的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: HDUOJ 6707 Shuffle C
- 下一篇: ADC0809转换器