Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

High cpu consumption low gpu consumption demo 3D inference #1198

Open
jamespoly862 opened this issue Jan 26, 2024 · 1 comment
Open

High cpu consumption low gpu consumption demo 3D inference #1198

jamespoly862 opened this issue Jan 26, 2024 · 1 comment

Comments

@jamespoly862
Copy link

jamespoly862 commented Jan 26, 2024

Demo 3D inference give me an error on gpu_device = self.device on webcam_detector.
Cancel this row, 3D inference function but give me high cpu consumption and low gpu consumption and too slowly.
I have a nvidia 3070 ti and 2D inference work. I don't know why... CUDA is correctly installed.

Kind regards,
James

@ano0904
Copy link

ano0904 commented Feb 21, 2024

I don't know is it related but I notice there is lots of torch tensor is using cpu instead of gpu even it is okay to use, one example is here, it turns the image from ndarray to torch tensor in cpu, but it is absolutely fine to use cuda tensor.

I changed it to use cuda tensor instead and already reduce a certain amount of cpu cost and speed up the process by a lot.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants