TensorRT
文章平均质量分 53
dragonfly91
这个作者很懒,什么都没留下…
展开
-
pycuda._driver.LogicError: cuDeviceGet failed: initialization error解决方法
在使用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问题解原创 2021-05-17 11:07:26 · 3800 阅读 · 3 评论 -
pth->onnx->trt踩坑笔记1
1、最近编译一个torch的pth转TensorRT的trt工具的时候遇到的错误:/usr/bin/ld: 找不到 -lnvinfer/usr/bin/ld: 找不到 -lcudartcollect2: error: ld returned 1 exit statusmake[2]: *** [CMakeFiles/dbnet.dir/build.make:120:dbnet] 错误 1make[1]: *** [CMakeFiles/Makefile2:95:CMakeFiles/db原创 2021-01-26 17:31:30 · 1988 阅读 · 3 评论 -
Tensorrt7.2.1安装问题libcudnn.so.8: cannot open shared object file: No such file or directory
最近安装TensorRT7.2.1,cuda版本10.2,cudnn8.02,按照官网教程安装完成之后,import tensorrt的时候遇到问题。ImportError: libcudnn.so.8: cannot open shared object file: No such file or directory我的cuda是做了一个软链接,软链接地址在/usr/local/cuda下,我首先将/usr/local/cuda/lib64的路径,添加到.bashrc里面,即export LD原创 2020-12-18 15:04:53 · 8411 阅读 · 3 评论