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