From 85fe9a7f670fe46fa84583e7a6da06e31f402b64 Mon Sep 17 00:00:00 2001 From: Victor Lin <13424970+victorlin@users.noreply.github.com> Date: Mon, 9 Oct 2023 11:25:03 -0700 Subject: [PATCH] Install jaxlib from PyPI on linux/arm64 MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit The PyPI release of jaxlib version 0.4.18 includes Linux aarch64 wheels¹. The current dependency definition in evofr² should allow this latest version to be pulled automatically. ¹ https://pypi.org/project/jaxlib/0.4.18/#files ² https://github.com/blab/evofr/blob/fa042389da267a7723f33dfaece9dd12f29dcaae/pyproject.toml#L13 --- Dockerfile | 16 ---------------- 1 file changed, 16 deletions(-) diff --git a/Dockerfile b/Dockerfile index fd11c23..04eaf2e 100644 --- a/Dockerfile +++ b/Dockerfile @@ -253,19 +253,6 @@ RUN apt-get update && apt-get install -y --no-install-recommends \ # 1. Install programs via pip -# Install jaxlib & jax on linux/arm64 -# jaxlib, an evofr dependency, does not have official pre-built binaries for -# linux/arm64. A GitHub user has provided them in a fork repo. -# https://github.com/google/jax/issues/7097#issuecomment-1110730040 -# Also hard-coding jax version here since it needs to match the jaxlib version -# The minimum version requirement for jaxlib is checked at runtime rather than by pip -# https://jax.readthedocs.io/en/latest/jep/9419-jax-versioning.html#how-are-jax-and-jaxlib-versioned -RUN if [[ "$TARGETPLATFORM" == linux/arm64 ]]; then \ - pip3 install https://github.com/yoziru/jax/releases/download/jaxlib-v0.4.6/jaxlib-0.4.6-cp310-cp310-manylinux2014_aarch64.manylinux_2_17_aarch64.whl \ - jax==0.4.6 \ - ; \ - fi - # Install envdir, which is used by pathogen builds RUN pip3 install envdir==1.0.1 @@ -358,9 +345,6 @@ RUN /builder-scripts/download-repo https://github.com/nextstrain/augur "$(/build && pip3 install --editable . # Add evofr for forecasting -# NOTE: if there is an issue with the evofr installation on linux/arm64, make -# sure to check that the jaxlib installation above satisfies the latest evofr -# dependency requirements. RUN pip3 install evofr # ———————————————————————————————————————————————————————————————————— #