-
Notifications
You must be signed in to change notification settings - Fork 4
Open
Description
Thank you for the nice wrapper. I managed to use the CUDA support with these steps if anyone else wants to do it as well:
- The latest version on PyPI (0.2) doesn't support the new Pythonic API, clone the repo and install an editable build. (
pip install -e .) - Download the latest OIDN binaries from here: https://github.com/RenderKit/oidn/releases
- Replace all the dll files and copy
OpenImageDenoise_device_cuda.dllas well. - In
__init__.py, add CUDA dll as wellctypes.CDLL(os.path.join(cur_path, f"lib.win.x64/OpenImageDenoise_device_cuda.dll"))
- Disable the
GetDeviceErrorfunction (it was raising astructrelated error on my end.) - In
Buffer.createfunction, change the tensor creation function to the current arguments:bf.buffer_delegate = torch.zeros(*storage_shape, dtype=torch.float32)
- And done! Thanks to this I'm using denoising close to realtime in my toy pathtracer project.
If I can find time I can just open a PR as well.
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels