已解决 ｜ trtexec: command not found， 在 Docker 容器中找不到 trtexec 命令

在下载的 TensorRT 文件夹中的 bin 路径下明明有 trtexec， 但报错找不到

报错如下

root@bb628c3eb276:/swinir-trt# trtexec --help
bash: trtexec: command not found

需要将 trtexec 所在的目录添加到环境变量 PATH 中。

1. 确定 trtexec 的完整路径：

find /usr -name trtexec

# 输出结果如下
./TensorRT-8.6.1.6/targets/x86_64-linux-gnu/bin/trtexec
./TensorRT-8.6.1.6/samples/trtexec

将上述结果再加上当前路径得到
/当前路径/TensorRT-8.6.1.6/targets/x86_64-linux-gnu/bin

2. 添加到 PATH 环境变量：

一种简单的方法是使用 export 命令。如果 trtexec 的目录是 /当前路径/TensorRT-8.6.1.6/targets/x86_64-linux-gnu/bin，则可以运行以下命令：

export PATH=$PATH:/swinir-trt/TensorRT-8.6.1.6/targets/x86_64-linux-gnu/bin

这会将 trtexec 的目录添加到当前会话的 PATH 环境变量中。如果退出容器或开启新的会话，这个更改将不会保留。

3. 持久化环境变量更改

为了使 PATH 的更改在容器重启后仍然有效，可以将 export 命令添加到容器的启动脚本或者 .bashrc / .profile 等 Shell 初始化文件中。例如，编辑 /root/.bashrc 文件，在文件末尾添加上述 export 命令：

echo "export PATH=\$PATH:/swinir-trt/TensorRT-8.6.1.6/targets/x86_64-linux-gnu/bin" >> /root/.bashrc

4. 应用更改

如果修改了 .bashrc 或 .profile 文件，可以通过运行如下命令来应用更改，或者重新登录到容器中以触发更改。

source /root/.bashrc 

最后再次输入 trtexec --help 测试有效！