本文介绍了一个关于使用Python API进行TensorRT推理时遇到的PyCUDA初始化错误及其解决方案。错误出现在创建设备上下文时，通过引入pycuda.autoinit模块解决了问题。
在使用Python API进行TensorRT推理时遇到了这个错误，出错的代码行，形如：
# Create a Context on this device,
self.cfx = cuda.Device(0).make_context()
报错为：pycuda._driver.LogicError: cuDeviceGet failed: initialization error
后来谷歌了一下，发现是import的时候少了一项，只要加上
import pycuda.autoinit
问题解决
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
