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

Any updates if it works with new versions of cuda 11.8? #304

Open
yemuzi opened this issue Aug 21, 2023 · 3 comments
Open

Any updates if it works with new versions of cuda 11.8? #304

yemuzi opened this issue Aug 21, 2023 · 3 comments

Comments

@yemuzi
Copy link

yemuzi commented Aug 21, 2023

on 2070 gpu card,
image

Cuda failure: out of memory
/root/trt8/src/Conv2d.cpp:177
Aborting...

@ckurtz22
Copy link

ckurtz22 commented Nov 5, 2023

I've tested it up to Cuda 12.3, works fine. The memory error is because the tests are written around running it on a Jetson Xavier with unified memory. For normal GPUs you need to comment out this line so when it can just serialize the network without running it: https://github.com/ceccocats/tkDNN/blob/master/src/Conv2d.cpp#L176
And then you can load the network.rt file and run it later on.

@vtyw
Copy link

vtyw commented Nov 22, 2023

@ckurtz22 What version of OpenCV did you have to use with CUDA 12?

@ckurtz22
Copy link

@ckurtz22 What version of OpenCV did you have to use with CUDA 12?

I used the default opencv installation for Ubuntu 20.04, I believe it was 4.1. I also was able to build 4.8 from source and get it working without issue. I have not been using OpenCV with CUDA enabled,

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

3 participants