Google Colab安装TensorRT遇问题：依赖已满足但命令未找到求助

问题原因 #

你通过pip install tensorrt安装的是TensorRT的Python绑定库，这个包不包含tensorrt命令行工具，所以直接运行!tensorrt -h会提示找不到命令。而且Google Colab其实已经预装了TensorRT环境，手动用pip安装可能会导致版本冲突。

正确操作步骤 #

• 确认Colab预装的TensorRT环境
  先运行以下命令查看已安装的TensorRT版本：

  import tensorrt
  print(tensorrt.__version__)
  

  如果能正常输出版本号，说明环境已经可用，不需要额外安装。

• 若需要更新或重新配置（可选）
  如果预装版本不符合需求，可以通过Colab的系统包管理器安装完整的TensorRT（包含命令行工具）：

  !apt-get update
  !apt-get install -y tensorrt
  

  安装完成后，运行以下命令验证命令行工具：

  !tensorrt -h
  

• 验证Python接口可用性
  不管是预装还是重新安装，都可以用以下代码测试TensorRT的Python接口是否正常工作：

  import tensorrt as trt
  import torch
  
  # 确认CUDA可用
  assert torch.cuda.is_available(), "GPU运行时未正确配置"
  
  # 创建TensorRT logger
  logger = trt.Logger(trt.Logger.WARNING)
  # 构建一个简单的引擎示例
  builder = trt.Builder(logger)
  network = builder.create_network()
  input_layer = network.add_input("input", trt.float32, (1, 3, 224, 224))
  print("TensorRT Python接口正常工作")
  

注意事项 #

• 不要同时用pip和apt-get安装TensorRT，避免版本冲突，优先使用Colab预装的环境。
• 如果你是为了运行Nougat项目，只需要确保TensorRT的Python接口可用即可，不需要命令行工具，Nougat会直接调用Python库。

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