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

Cannot import JIT optimized kernels. Did you forget to install Custom Kernel Extension? #180

Open
Alex-Songs opened this issue Aug 7, 2022 · 1 comment

Comments

@Alex-Songs
Copy link

Cannot import JIT optimized kernels. Did you forget to install Custom Kernel Extension?

@ghostplant
Copy link
Contributor

It is usually due to environment issues (e.g. improper CXX compiler, CUDA dependencies, or NCCL dependencies) which make Tutel installation only enable CPU support, e.g. python3 -m tutel.examples.helloworld --device cpu.

The common way to enable GPU support correctly is to see carefully about the log during the setup of Tutel, find the dependency error information in the log and fix them up. If you are not quite familiar with certain errors, you can share the installation log here which would help us to diagnose.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

No branches or pull requests

2 participants