NVIDIA® TensorRT™ is an SDK for high-performance deep learning inference on NVIDIA GPUs. This repository contains the open source components of TensorRT.
I try to export yolov5s model as engine in WSL2, it seems like I have finished, but when I try to calculate mAP, It returns zeros. So I guess something wrong in WSL2
Environment
TensorRT Version: 8.2.2.1
NVIDIA GPU: rtx3070 laptop
NVIDIA Driver Version: 510.06
CUDA Version: 11.43
CUDNN Version: 8.2.4
Operating System: win11
Python Version (if applicable): python38
Tensorflow Version (if applicable):
PyTorch Version (if applicable): torch1.10.1
Baremetal or Container (if so, version):
lix19937
commented
4 months ago
Description
I try to export yolov5s model as engine in WSL2, it seems like I have finished, but when I try to calculate mAP, It returns zeros. So I guess something wrong in WSL2
Environment
TensorRT Version: 8.2.2.1 NVIDIA GPU: rtx3070 laptop NVIDIA Driver Version: 510.06 CUDA Version: 11.43 CUDNN Version: 8.2.4 Operating System: win11 Python Version (if applicable): python38 Tensorflow Version (if applicable): PyTorch Version (if applicable): torch1.10.1 Baremetal or Container (if so, version):
Relevant Files
https://github.com/ultralytics/yolov5/blob/master/export.py
Steps To Reproduce