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

Stop installing GPU dependencies #600

Closed
pquentin opened this issue Sep 12, 2023 · 0 comments · Fixed by #615
Closed

Stop installing GPU dependencies #600

pquentin opened this issue Sep 12, 2023 · 0 comments · Fixed by #615

Comments

@pquentin
Copy link
Member

Installing all Eland dependencies locally or in the Docker image installs 4.2GB of dependencies, the two largest being PyTorch (1.9GB) and the CUDA dependencies (https://pypi.org/project/nvidia-cudnn-cu11/, https://pypi.org/project/nvidia-cublas-cu11/, https://pypi.org/project/nvidia-cuda-nvrtc-cu11/ and https://pypi.org/project/nvidia-cuda-runtime-cu11/, taking 1.4GB).

Since we don't support GPUs, installing those is wasteful, and we should install the PyTorch variant that only supports CPUs:

pip install torch==1.13.1+cpu --extra-index-url https://download.pytorch.org/whl/cpu

It's not possible to specify the extra index URL when doing python -m pip install eland[pytorch], but we could instead ask to run the above command. In any case, it's an easy fix in the Docker image which is probably the main way that PyTorhc is used with Eland today (and #407 will make this even more true).

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

Successfully merging a pull request may close this issue.

1 participant