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

slow denoise Ubuntu 22.04 #159

Open
stevelcb opened this issue Jun 1, 2024 · 3 comments
Open

slow denoise Ubuntu 22.04 #159

stevelcb opened this issue Jun 1, 2024 · 3 comments

Comments

@stevelcb
Copy link

stevelcb commented Jun 1, 2024

Hi everyone
GraX 3.0.2 using AI denoise 3.0.1
Beelink Ryzen 7, 5800 8c/16t with Radeon graphics

A 6000x4000 pixel image takes over 30 minutes to denoise.
I've tried binning the image, inference batch sizes of 1, 2 and 4 and with/without hardware acceleration checked but without improvement in denoising speed.
AI 1.0.0 denoises the same image in around 5 minutes and background extraction using AI 1.0.1 is almost instantaneous.

Any ideas?
Cheers and thanks for a great project.

@schmelly
Copy link
Collaborator

schmelly commented Jun 2, 2024

Hi Steve,

hardware acceleration on Linux is available via onnxruntime and CUDA only for our pre-build package (afaik this requires an NVIDIA card). As onnxruntime does not provide pre-build rocm binaries at the moment unfortunately we cannot provide a pre-build GraXpert package for amd hardware either currently. Your only option would probably be to run GraXpert from source and build onnxruntime by yourself.

For completeness, the current Linux dependencies are:

  • CUDA 12
  • libcudnn8
  • a CUDA 12 compatible kernel module

CS, David

@stevelcb
Copy link
Author

stevelcb commented Jun 3, 2024

Hi and thanks for the prompt reply.
If I build from source with onnxruntime, will that give me gpu acceleration on amd?
I've had a look but it doesn't look trivial. If anyone has a recipe/method for this, do please share.
Cheers

@schmelly
Copy link
Collaborator

Hi Steve,

according to the installation instructions of onnxruntime, there is no binary distribution of it for AMD. That is, in my understanding, you would have to build onnxruntime first (I don't have experience with this step) and use that for starting GraXpert from sources. If you can build onnxruntime successfully and all driver dependencies are met it should work for you.

CS, David

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