search

dattalab / keypoint-moseq

https://keypoint-moseq.readthedocs.io
Other
63 stars 25 forks source link

cuSolver internal error #107

Closed lihao881230 closed 7 months ago

lihao881230 commented 8 months ago

Hi all,

I am also running into the cuSolver internal error on Windows 11 (CUBA 11.1 and cuDNN 8.2), which is pretty much described in issues #69 and #103 . I have tried both conda install and pip install and they gave me the same error. I was wondering if anyone has found a solution to this.

image

image

calebweinreb commented 8 months ago

I think the problem is that JAX isn't currently supported on Windows 11. Looking at issue https://github.com/dattalab/keypoint-moseq/issues/103, it seems like the best workaround for now is to use WSL2.

KonradDanielewski commented 7 months ago

I was testing some solutions to this issue and seems that it's just related to the cudnn and cudatoolkit installation in the env. I will have a fully tested solution later today and will write the specific versions and commands here - I just wanna test in on another computer also to see if it's not a fluke and maybe I just did something more than I think - I'm still running into OOMs (I think, jax seems to fail silently when it OOMs and just leads to kernel death)

calebweinreb commented 7 months ago

Thanks for working on this Konrad!

KonradDanielewski commented 7 months ago

Should be fixed with #110

calebweinreb commented 7 months ago

https://github.com/dattalab/keypoint-moseq/pull/110 is now merged into main. Thanks @KonradDanielewski !