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

installing on aarch64 #105

Open
jonglees opened this issue Nov 23, 2024 · 6 comments
Open

installing on aarch64 #105

jonglees opened this issue Nov 23, 2024 · 6 comments
Labels
help wanted Extra attention is needed

Comments

@jonglees
Copy link

Everything installs on AArch64 architecture,, but the two binaries don't run ( dssp and the DalphaBall )
I think there's a pure python version of dssp that could be used to replace the binary quite easily.

@EricDeveaud
Copy link

see: #80

@AssmannG
Copy link

AssmannG commented Dec 4, 2024

Hi , I am just trying to build BindCraft on a GraceHopper aarch64 platform.It actually fails for me already with building the conda env...

I used : bash install_bindcraft.sh --cuda '12.4' --pkg_manager 'conda'
The error it throws is:

PackagesNotFoundError: The following packages are not available from current channels:

  - jaxlib=[build=*cuda*]

I tried different cuda versions etc, always fails. Did you change something in the installation script for the jax/jaxlib packages/ channels/cuda versions? Or did it just work out of the box ?

Also , doing the installation on an linux-64 platform worked well.

Any help appreciated,
Greta

@martinpacesa
Copy link
Owner

Hi there, unfortunately I don't have a space to test this sort of environment, maybe some kind user can help out here :)

@martinpacesa martinpacesa added the help wanted Extra attention is needed label Dec 5, 2024
@AssmannG
Copy link

AssmannG commented Dec 6, 2024

Thanks! Yes, I was hoping @jonglees could maybe comment on that!

@EricDeveaud
Copy link

Hello,

maybee you can try to manually pick jax wheels from here https://storage.googleapis.com/jax-releases/jax_cuda_releases.html

I imagine something like that, let say you are using python version 3.10, eg

wget https://storage.googleapis.com/jax-releases/cuda12/jaxlib-0.4.27+cuda12.cudnn89-cp310-cp310-manylinux2014_aarch64.whl
pip install  ./jaxlib-0.4.27+cuda12.cudnn89-cp310-cp310-manylinux2014_aarch64.whl

@AssmannG
Copy link

AssmannG commented Dec 9, 2024

thanks, I will try to play around!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
help wanted Extra attention is needed
Projects
None yet
Development

No branches or pull requests

4 participants