PyTorch 转 TensorRT:Python 和 C++ 端环境搭建

一、配置TensorRT

  1. 解压安装文件:
cd ~/0-software/
tar xzvf TensorRT-8.6.1.6.Linux.x86_64-gnu.cuda-11.8.tar.gz
  1. 运行sudo gedit ~/.bashrc,在文档中添加:
export PATH=/home/wangbo/0-software/TensorRT-8.6.1.6/bin:$PATH
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/home/wangbo/0-software/TensorRT-8.6.1.6/lib

二、配置PyTorch

  1. 新建conda环境
conda create -n base_113
conda activate base_113
  1. 安装PyTorch 1.13.1
conda install pytorch==1.13.1 torchvision==0.14.1 pytorch-cuda=11.6 -c pytorch -c nvidia
  1. 安装pycuda
    去whl下载网站,根据cuda和python确定版本号,例如:2022.1
    然后运行安装命令:
pip install cuda-python
pip install pycuda == 2022.1
  1. 安装TensorRT
pip install tensorrt==8.6.1
posted @   仙海寻波  阅读(88)  评论(0编辑  收藏  举报
编辑推荐:
· 在.NET Core中使用异步多线程高效率的处理大量数据
· 聊一聊 C#前台线程 如何阻塞程序退出
· 几种数据库优化技巧
· 聊一聊坑人的 C# MySql.Data SDK
· 使用 .NET Core 实现一个自定义日志记录器
阅读排行:
· 字节豆包,来园广告
· 我用cursor, 半就开发了一个手机壁纸小程序,真的太强了
· 聊一聊 C#前台线程 如何阻塞程序退出
· 订单超时自动取消,我们是这样做的。。。
· C#字符串拼接的几种方式及其性能分析对比
点击右上角即可分享
微信分享提示