Andrew Alberding comments

Repositories
Issues
Comments

Results 4 comments of


                                            Andrew Alberding

cannot import name '_cext' !!

Also have the problem in GPUTreeExplainer - I compiled from source using 'python setup.py install' and it found CUDA... seemed to install fine. C extension was not built during install!...

Torch is not able to use GPU; add --skip-torch-cuda-test to COMMANDLINE_ARGS variable to disable this check

But I thought it would work in Windows even with this ROCM pytorch? Guess I'll have to switch to Linux.. pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/rocm5.1.1

GPUTreeExplainer raises import error on AWS Deep Learning AMI 44

I have the same issue - only with GPUTree. The build seems fine just like yours. Any solution?

GPUTreeExplainer raises import error on AWS Deep Learning AMI 44

I solved it by doing 'pip uninstall shap' twice, downloading a fresh tar release from this github, and then making sure I had gcc and g++ that matched the CUDA...