环境:
ubuntu18.04
pycuda
pil
tensorflow-gpu=1.14.0
cuda=10.0
cudnn=7.6.5
python=3.6
1.安装
按照大佬给出的方法安装:https://blog.csdn.net/u011681952/article/details/105973996
需要注意的是,tensorRT的deb安装方式要求CUDA也是deb格式安装的,否则容易出错(很多安装CUDA采用的是runfile格式)。
因我采用的安装CUDA方式时runfile方式,故我选择了local安装方法。
2.测试
按照大佬的《3.使用》步骤进行了测试,在3.3步碰到了**Could not find x.pgm. Searched in data paths: [’/usr/src/tensorrt/data’]
**错误,观察bug发现是common.py中locate_files的文件路径找不到。
3.解决方法
1.找到TensorRT的解压(安装)路径,我的安装路径为/home/(user name)/下载/TensorRT-7.0.0.11/
2.到该文件夹下的samples/python文件夹找到common.py,修改kDEFAULT_DATA_ROOT。
本人的测试图片,即.pgm文件保存在
/home/(user name)/下载/TensorRT-7.0.0.11/data/mnist
中,故将
kDEFAULT_DATA_ROOT = os.path.join(os.sep, “home”, “(user name)”, “下载”, “TensorRT-7.0.0.11”, “data”)
再保存即可。
如此方法不行,也可参考大佬的做法。