WSL2中TAO Toolkit TensorFlow容器执行TAO命令提示“未找到”求助

问题原因 #

TAO Toolkit 4.x版本的容器默认未将tao命令的路径加入系统全局PATH，直接输入tao会导致系统找不到该命令。

解决方法 #

方法1：直接使用完整路径执行命令 #

将原命令中的tao替换为容器内TAO命令的完整路径：

/opt/nvidia/tao/tao-toolkit/bin/tao deploy gazenet gen_trt_engine \
  -m /workspace/model_dir/gazenet_deployable_v1.0/model.etlt \
  -k nvidia_tlt \
  --data_type fp16 \
  --engine_file /workspace/model_dir/gazenet_deployable_v1.0/gazenet_engine.engine

方法2：将TAO路径添加到系统PATH #

先执行以下命令将TAO的bin目录加入PATH：

export PATH=/opt/nvidia/tao/tao-toolkit/bin:$PATH

之后即可直接使用你的原tao deploy命令。

方法3：激活容器内的虚拟环境（部分版本适用） #

如果容器内存在TAO的虚拟环境，先激活环境再执行命令：

source /opt/nvidia/tao/tao-toolkit/venv/bin/activate

激活后直接运行原tao deploy命令即可。

内容的提问来源于stack exchange，提问作者Novice