wesselb / neuralprocesses.jl Goto Github PK

I've been trying to move some intensive neural process work from python to julia, but this package doesn't seem to support the latest version of Stheno's kernel functions (i.e. Matern52 no longer exists, in favor of using Matern52Kernel).

I'd like to get some of the examples of this package working, and was curious what version of Stheno was used in the given examples? Additionally, are there plans to update the package to work with the latest version of the dependencies?

Thanks for all your work!

I have tried to have a quick start by running the train.jl. However, it end up with an error

ERROR: LoadError: UndefVarError: CUDNN not defined

The code run in the environment:

• Windows 11
• Nvidia RTX a1000 laptop gpu (driver version 536.67)
• Julia 1.9.2
• CUDA Toolkit 12.2

Thank you for your help.