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

CPU only package #87

Open
hadim opened this issue Oct 24, 2022 · 5 comments
Open

CPU only package #87

hadim opened this issue Oct 24, 2022 · 5 comments
Labels
question Further information is requested

Comments

@hadim
Copy link
Member

hadim commented Oct 24, 2022

Comment:

Would that be possible to get a CPU only package similar to pytorch-cpu for example? I am deploying a backend with docker and one of my direct dep depends on openmm and cudatoolkit is selected when installing the conda env and it increases the docker image size significantly (from 3GB to 5GB).

Being able to specify openmm-cpu in my env file would helpful xD

@hadim hadim added the question Further information is requested label Oct 24, 2022
@peastman
Copy link
Contributor

At present we don't have any plans to create a CPU only conda package. But we're hoping to provide PyPI packages in the future, which will give a route for installing without needing to include a full CUDA toolkit.

@hadim
Copy link
Member Author

hadim commented Oct 24, 2022

Thanks @peastman. I am trying to stay away from pip package when possible so a CPU only conda package would be really nice, but I also understand the added burden to build it!

@peastman
Copy link
Contributor

Though it's not an ideal solution, you can force it to remove the CUDA toolkit after everything has been installed with

conda remove --force cudatoolkit

You probably also want to run conda clean --all. That will minimize the size of your final image.

@hadim
Copy link
Member Author

hadim commented Oct 25, 2022

Indeed not ideal but clearly a nice hotfix. Thanks!

@mattwthompson
Copy link
Member

ref/tag openmm/openmm#3059

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

No branches or pull requests

3 participants