Closed pikaqkio closed 4 months ago
默认是编译并打包成python10的whl,你看一下trtyolo虚拟环境的版本是python10吗?
默认是编译并打包成python10的whl,你看一下trtyolo虚拟环境的版本是python10吗?
(trtyolo) C:\Windows\system32>python --version
Python 3.10.6
@pikaqkio 现在好了吗?
@pikaqkio 现在好了吗?
之前系统同时安装了CUDA v11.8和v12.1,即使在系统的PATH环境变量中将CUDA v11.8路径置顶也不行。 现在只保留CUDA 12.1就可以了,不知道是不是这个原因导致的。 还有顺利提一嘴,我安装cuDNN时候没有zlibwapi也是可以的(原因是没找到这个文件的下载。。) 感谢指导!
(trtyolo) C:\Yolo\TensorRT-YOLO\demo\detect>trtyolo infer -e models/yolov8n-0508.engine -i images -o output -l labels.txt --cudaGraph
[I] Successfully found necessary library paths:
{
"cudart": "C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v12.1\\bin",
"nvinfer": "C:\\Program Files\\NVIDIA GPU Computing Toolkit\\TensorRT\\v8.6.1.6\\lib",
"cudnn": "C:\\Program Files\\NVIDIA GPU Computing Toolkit\\cnDNN\\v8.9.7.29\\bin"
}
[I] Infering data in images
Processing batches ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 100% 0:00:02
[S] Benchmark results include time for H2D and D2H memory copies, preprocessing, and postprocessing.
CPU Average Latency: 1.046 ms
GPU Average Latency: 1.029 ms
Finished Inference.
(trtyolo) C:\Yolo\TensorRT-YOLO\demo\detect>
感谢反馈,大概率是多版本CUDA导致的问题。zlib
在这个项目上是用不上的,但是有时使用torch
训练却是必须的。
感谢反馈,大概率是多版本CUDA导致的问题。
zlib
在这个项目上是用不上的,但是有时使用torch
训练却是必须的。
题外话,咨询下大佬cv2.waitKey的问题,在OpenCV中,cv2.waitKey()的最小间隔时间是16ms吗,我发现哪怕是cv2.waitKey(1)最终显示的fps也无法超过60fps。。
关于 cv2.waitKey
的问题你可以参考这个回答。https://github.com/laugh12321/TensorRT-YOLO/issues/19#issuecomment-2085189425
CLI和python方式推理都报了相同错误,前面的步骤都是没问题的。