在使用tensorrt调用已经生成好的.engine模型,但是需要pycuda库,pip安装屡屡失败
尝试过类似方法都不行,换numpy,settools版本也失败
后来发现是python版本不对,之前conda虚拟环境都是python=3.8
conda create -n [name] python=3.9
pip install pycuda
这样就行了
在使用tensorrt调用已经生成好的.engine模型,但是需要pycuda库,pip安装屡屡失败
尝试过类似方法都不行,换numpy,settools版本也失败
后来发现是python版本不对,之前conda虚拟环境都是python=3.8
conda create -n [name] python=3.9
pip install pycuda
这样就行了