- 博客(9)
- 收藏
- 关注
RSS订阅原创 第三步：onnx模型导入tensorrt生成优化engine + 在GPU上推理
一、环境Ubuntu18.04 + cuda10.0 + python3.7.6 + pytorch1.4.0 + tensorrt7.0.0.11二、onnx模型导入tensorrt生成优化engine找到TensorRT的安装目录，里面有一个trtexec工具，完整路径：/path/to/TensorRT-7.0.0.11/bin/trtexec。这个工具可以直接生成引擎，不需要写代码，终端命令：# --onnx: onnx模型路径；--saveEngine: 将要保存engine的路径；--v
2021-05-31 15:57:07
4156
1
原创 第二步：pytorch模型转onnx模型步骤和可能遇到的问题
一、本地(win10)环境cuda11.0 + python3.7.4 + pytorch1.7.1 + onnx1.9.0二、可能出现的问题1. pytorch与tensorrt版本对应问题pytorch转onnx模型是为了后续导入tensorrt做准备，所以要考虑pytorch与tensorrt版本对应的问题。由于服务器的环境是cuda10.1 + tensorrt7.0.0.11，所以要先把pytorch模型从1.7.1降到1.4.0。以efficientnet_b1为例，先在pytorch1
2021-05-25 20:24:12
9480
4
原创 第一步：ubuntu18.04装TensorRT8.0.0.3 + onnx1.8.0 + onnx_tensorrt
一、环境说明Ubuntu18.04 + cuda10.0 + python3.7.6 + pytorch170我是直接在镜像里配置所有环境，所以默认root权限。如果在本地配置环境时报权限不足的错误，请自行在语句前加上 sudo。二、安装TensorRT7.0.0.11下载安装包 tar包官网下载解压 tar xzvf TensorRT-7.0.0.11.Ubuntu-18.04.x86_64-gnu.cuda-10.0.cudnn7.6.tar.gz配置环境变量 vim ~/.ba
2021-05-24 16:27:57
3264
2
原创 戴尔电脑安装ubuntu18.04+win10双系统
我的电脑型号：Dell XPS 8940此文档仅用于记录个人的安装过程，不同电脑配置的差别可能导致我的方法不适用于其他电脑，仅供参考。为避免操作失误带来的后果，一定一定一定一定一定一定要先把Windows系统中所有的重要文件备份备份备份备份备份备份，以防万一。一、官网下载Ubuntu18.04 LTSUbuntu18.04 LTS下载官网用迅雷下会快很多。Ubuntu不要用最新版本，不稳定，Ubuntu18.04就挺好，Ubuntu16.04也不错。二、制作U盘启动盘1. 下载rufus用ru
2020-08-10 18:06:43
6848
6
原创 Docker新建容器
一、查看docker中的镜像先查看docker中有哪些镜像，决定在哪个镜像中创建容器（如果没有镜像，要先拉个镜像，一般Ubuntu16.04或Ubuntu14.04就可以了）。查看镜像命令：sudo docker images会列出docker中所有的镜像，比如我电脑中有：REPOSITORY TAG IMAGE ID CREATED SIZEubuntu
2020-06-16 14:21:06
1393
原创 Ubuntu下配置Visual Studio Code的C++自动补全功能
Visual Studio Code是一个非常好用的编辑器，但是它没有C++自动补全功能，这让我们在编程的时候很不方便，以下是Ubuntu系统下VS Code的C++自动补全配置方法。一、安装C++相关依赖包和插件sudo apt install gccsudo apt install gdbsudo apt install clangsudo apt install lldb二、在应用商店中安装C/C++扩展快捷键Ctrl+Shift+X打开扩展，在VS Code界面左上角输入 C++，选
2020-05-26 17:01:18
3574
原创 Docker服务器无法show图片解决方法（连到本地服务器上显示）
Docker服务器没有桌面环境，所以要把Docker中程序显示的图片连到本地服务器上显示。一、查看本地服务器的ip在本地终端输入：ifconfig记住自己本地服务器的ip，就是inet addr字段后面的一串数。比如我的是172.17.0.1。二、在Docker容器中的操作在Docker终端输入：root@ba14531badf8:/# export DISPLAY=172.17....
2020-04-22 14:40:19
2428
原创 解决TensorFlow只能保存5个模型的问题
直奔主题在训练模型的代码中找到这句代码：tf.train.Saver（），改成：tf.train.Saver(max_to_keep = m) # m为你想保存的模型数量扩展Saver类中的可选参数tf.train.Saver(max_to_keep = m, keep_checkpoint_every_n_hours = n)max_to_keep：保存离当前训练最近的模型数量，默...
2019-12-10 13:59:45
1783
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人