<p dir="auto">Hi! I can install jax but not jaxlib on debian stretch with python3.9.</p>
<p dir="auto">What version do I need to get this to work? Is it even possible?</p>
<p dir="auto">Thanks for any help!!</p>
<p dir="auto">Dockerfile:</p>
<div class="snippet-clipboard-content notranslate position-relative overflow-auto" data-snippet-clipboard-copy-content="RUN echo &quot;Installing dependencies...&quot; &amp;&amp; \
    apt-get update &amp;&amp; apt-get install -y \
    build-essential \
    git
WORKDIR /usr/src/app
COPY requirements.txt .
RUN python3 -m pip install --upgrade pip
RUN python3 -m pip install --upgrade Pillow
RUN pip3 install -r requirements.txt
RUN pip3 install --upgrade &quot;jax[cuda]&quot; -f https://storage.googleapis.com/jax-releases/jax_releases.html
COPY . .
CMD sleep infinity"><pre class="notranslate"><code class="notranslate">RUN echo "Installing dependencies..." &amp;&amp; \
    apt-get update &amp;&amp; apt-get install -y \
    build-essential \
    git
WORKDIR /usr/src/app
COPY requirements.txt .
RUN python3 -m pip install --upgrade pip
RUN python3 -m pip install --upgrade Pillow
RUN pip3 install -r requirements.txt
RUN pip3 install --upgrade "jax[cuda]" -f https://storage.googleapis.com/jax-releases/jax_releases.html
COPY . .
CMD sleep infinity
</code></pre></div>
<p dir="auto">I get this error:</p>
<div class="snippet-clipboard-content notranslate position-relative overflow-auto" data-snippet-clipboard-copy-content="ERROR: Could not find a version that satisfies the requirement jaxlib (from versions: none)
ERROR: No matching distribution found for jaxlib"><pre class="notranslate"><code class="notranslate">ERROR: Could not find a version that satisfies the requirement jaxlib (from versions: none)
ERROR: No matching distribution found for jaxlib
</code></pre></div>
<div class="snippet-clipboard-content notranslate position-relative overflow-auto" data-snippet-clipboard-copy-content="The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File &quot;/usr/src/app/app.py&quot;, line 8, in &lt;module&gt;
    import jax
  File &quot;/usr/local/lib/python3.9/site-packages/jax/__init__.py&quot;, line 37, in &lt;module&gt;
    from . import config as _config_module
  File &quot;/usr/local/lib/python3.9/site-packages/jax/config.py&quot;, line 18, in &lt;module&gt;
    from jax._src.config import config
  File &quot;/usr/local/lib/python3.9/site-packages/jax/_src/config.py&quot;, line 27, in &lt;module&gt;
    from jax._src import lib
  File &quot;/usr/local/lib/python3.9/site-packages/jax/_src/lib/__init__.py&quot;, line 39, in &lt;module&gt;
    raise ModuleNotFoundError(
ModuleNotFoundError: jax requires jaxlib to be installed. See https://github.com/google/jax#installation for installation instructions."><pre class="notranslate"><code class="notranslate">The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/usr/src/app/app.py", line 8, in &lt;module&gt;
    import jax
  File "/usr/local/lib/python3.9/site-packages/jax/__init__.py", line 37, in &lt;module&gt;
    from . import config as _config_module
  File "/usr/local/lib/python3.9/site-packages/jax/config.py", line 18, in &lt;module&gt;
    from jax._src.config import config
  File "/usr/local/lib/python3.9/site-packages/jax/_src/config.py", line 27, in &lt;module&gt;
    from jax._src import lib
  File "/usr/local/lib/python3.9/site-packages/jax/_src/lib/__init__.py", line 39, in &lt;module&gt;
    raise ModuleNotFoundError(
ModuleNotFoundError: jax requires jaxlib to be installed. See https://github.com/google/jax#installation for installation instructions.
</code></pre></div>
<p dir="auto">Potentially related to <a class="issue-link js-issue-link" data-error-text="Failed to load title" data-id="916640295" data-permission-text="Title is private" data-url="https://github.com/google/jax/issues/6932" data-hovercard-type="issue" data-hovercard-url="/google/jax/issues/6932/hovercard" href="https://github.com/google/jax/issues/6932">#6932</a> or <a class="issue-link js-issue-link" data-error-text="Failed to load title" data-id="927516918" data-permission-text="Title is private" data-url="https://github.com/google/jax/issues/7052" data-hovercard-type="issue" data-hovercard-url="/google/jax/issues/7052/hovercard" href="https://github.com/google/jax/issues/7052">#7052</a>.</p>
<p dir="auto">I am using a heterogeneous cluster with various different compute options. I got Jax and Flax installed fine on the CPU system. Then I tried using a GPU node (specifically dual socket Cavium ThunderX2 99xx 32-core processors with two NVIDIA Tesla V100 GPUs), but am unable to install jaxlib:</p>
<div class="snippet-clipboard-content notranslate position-relative overflow-auto" data-snippet-clipboard-copy-content="$ pip install --upgrade jax jaxlib==0.1.68+cuda111 -f https://storage.googleapis.com/jax-releases/jax_releases.html
Looking in links: https://storage.googleapis.com/jax-releases/jax_releases.html
Requirement already satisfied: jax in /home/southworth/.conda/envs/jax_arch/lib/python3.7/site-packages (0.2.16)
ERROR: Could not find a version that satisfies the requirement jaxlib==0.1.68+cuda111 (from versions: none)
ERROR: No matching distribution found for jaxlib==0.1.68+cuda111"><pre class="notranslate"><code class="notranslate">$ pip install --upgrade jax jaxlib==0.1.68+cuda111 -f https://storage.googleapis.com/jax-releases/jax_releases.html
Looking in links: https://storage.googleapis.com/jax-releases/jax_releases.html
Requirement already satisfied: jax in /home/southworth/.conda/envs/jax_arch/lib/python3.7/site-packages (0.2.16)
ERROR: Could not find a version that satisfies the requirement jaxlib==0.1.68+cuda111 (from versions: none)
ERROR: No matching distribution found for jaxlib==0.1.68+cuda111
</code></pre></div>
<p dir="auto">or even</p>
<div class="snippet-clipboard-content notranslate position-relative overflow-auto" data-snippet-clipboard-copy-content="pip install jaxlib
ERROR: Could not find a version that satisfies the requirement jaxlib (from versions: none)
ERROR: No matching distribution found for jaxlib"><pre class="notranslate"><code class="notranslate">pip install jaxlib
ERROR: Could not find a version that satisfies the requirement jaxlib (from versions: none)
ERROR: No matching distribution found for jaxlib
</code></pre></div>
<p dir="auto">Below is my pip version, which is up to date. Pip also installs other libraries fine (numpy, scipy, etc.)</p>
<div class="snippet-clipboard-content notranslate position-relative overflow-auto" data-snippet-clipboard-copy-content="$ pip --version
pip 21.1.2 from /home/southworth/.conda/envs/jax_arch/lib/python3.7/site-packages/pip (python 3.7)"><pre class="notranslate"><code class="notranslate">$ pip --version
pip 21.1.2 from /home/southworth/.conda/envs/jax_arch/lib/python3.7/site-packages/pip (python 3.7)
</code></pre></div>