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

REQ: pre compiled wheel file for torchlars package #8

Open
ghostcow opened this issue Jun 22, 2020 · 2 comments
Open

REQ: pre compiled wheel file for torchlars package #8

ghostcow opened this issue Jun 22, 2020 · 2 comments

Comments

@ghostcow
Copy link

ghostcow commented Jun 22, 2020

Is it possible to pre-compile the package (make a wheel) for this repository?
I need to install it on a computer without GPUs and it won't let me.

@ghostcow ghostcow reopened this Oct 5, 2020
@ghostcow ghostcow changed the title REQ: pip package for torchlars REQ: pre compiled wheel file for torchlars package Oct 5, 2020
@chiheonk
Copy link

chiheonk commented Oct 7, 2020

Unfortunately, torchlars is only supported for machines with CUDA GPUs. It is specifically designed to speed up the computations in LARS optimizer by combining them into one GPU kernel.

@ghostcow
Copy link
Author

ghostcow commented Oct 7, 2020

Thank you for your amazing work!
That is understood, but my situation requires installing (not using) the package on a server with no GPU.
If you had a .whl file, or dropped the GPU check on compilation I could just compile it with CUDA. Similar to how one can install the GPU-enabled version of pytorch even without using the GPU functions.

Is it possible to add this feature?

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

No branches or pull requests

2 participants