Comments (5)
wang-xinyu
commented on May 28, 2024
试下FP32
或者试试TensorRT 8.4
from tensorrtx.
B1SH0PP
commented on May 28, 2024
试下FP32 或者试试TensorRT 8.4
您好! 很高兴真的能收到您的回复!
改为FP32是只要改这里吗?
我目前改好后, 重新编译, 重新生成了wts和engine文件, 测试下来依然没有效果.
改tensorRT 8.4x的版本可能会有点麻烦, 我看我目前装的cuda11.7不支持tensorRT8.4x版本, 这可能会有点耗时, 我待会儿测试下, 我不清楚cuda是否会有版本冲突, 是否得卸载干净.
from tensorrtx.
B1SH0PP
commented on May 28, 2024
from tensorrtx.
wang-xinyu
commented on May 28, 2024
如果是1060，那加下61 arch试试
set(CMAKE_CUDA_ARCHITECTURES 61 75 86 89)
from tensorrtx.
B1SH0PP
commented on May 28, 2024
@wang-xinyu 成功了! 太有实力了哥!
from tensorrtx.
Related Issues (20)
- yolov5 nms支持CUDA加速 HOT 5
- ImportError: libnvinfer.so.8: cannot open shared object file: No such file or directory HOT 3
- cmake error HOT 1
- The recently updated YOLOv8-Pose does not come with dedicated code for converting to the .wts format. HOT 2
- yolov8-pose似乎 不支持自己训练的模型 HOT 7
- kernel weights has count 2688 but 32640 was expected HOT 1
- yolov5编译问题 HOT 2
- yolov5s.pt微调后在生成.wts报错，改了数据类别，但类别总数不变。麻烦大大帮忙看看~谢谢！！new.pt是我微调后的模型 HOT 1
- ModuleNotFoundError: No module named 'utils.torch_utils' HOT 2
- 博主你好，这个能部署到windows上面了吗 HOT 6
- win10 compile yolov8 happen:yololayer.cu(262): error : expression must have a constant value HOT 12
- 不使用cmake搭建环境 HOT 21
- yolov8-seg, output error HOT 18
- YOLOV8 Assertion `serialized_engine' failed. 使用自己的.pt 生成 .wts 是成功的，但转.engine 是报错。chat说是卷积层权重数量不匹配 HOT 1
- Unable to build engine file for Yolov7 Custom model
- How to use file yolov8_det_trt.py to check one's own model
- yolov5-7.0，about anchor
- 可以使用onnx模型转换为tensorrt模型，再进行推理吗
- the command : ./yolov5_det -s yolov5s.wts yolov5s.engine s, error!!!
- widonws下yolov8模型生成wts转engine文件报错
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from tensorrtx.