Не удалось установить jaxlib

Question

Я пытаюсь установить jaxlib на свои окна 10 с помощью следующей команды, которую я нашел в документации.

pip install jaxlib

Он показывает следующую ошибку

Collecting jaxlib
  Could not find a version that satisfies the requirement jaxlib (from versions: None)
No matching distribution found for jaxlib

Кто-нибудь может мне помочь, заранее спасибо!

Answer 1

Jaxlib не поддерживается в Windows, вы можете увидеть его здесь.. https://github.com/google/jax/issues/438

Answer 2

Возможно, я что-то упустил, но благодаря jax - windows-builder от Cloudhan теперь довольно легко установить JAX и jaxlibна окнах. Например

      pip install jax==0.3.13 https://whls.blob.core.windows.net/unstable/cuda111/jaxlib-0.3.7+cuda11.cudnn82-cp38-none-win_amd64.whl

Как там объяснялось , мне пришлось скопировать адрес ссылки jaxlib интересующего меня файла whl , т.е. https://whls.blob.core...выше).

Но я также должен был позаботиться о том, чтобы версия JAX и версия Jaxlib были совместимы, совместимость которых легко оценить на github.com/google/jax/releases : просто выберите последние два из которых (!) ^{номера версий! нечего оттуда скачивать.}

---

проверено с версиями, указанными выше . _{Т.е. python3.8-64 и jax==0.3.13 и jaxlib-0.3.7}

Answer 3

Я сам проходил через этот процесс последние два дня, поэтому вот что я сделал:

1. скачать и установить последнюю версию Microsoft Visual Studio (чтобы получить компилятор c ++ и набор инструментов)

2. скачать и установить python

3. создать виртуальную среду Python с пакетами pip, setuptools, six, wheel и numpy (я сделал это в графическом интерфейсе pycharm)

4. скачать jax

5. откройте Windows PowerShell от имени администратора, перейдите в каталог jax и выполните следующие шаги (команды заключены в кавычки)

• установить шоколадный (менеджер пакетов для простой установки bazel)

Set-ExecutionPolicy Bypass -Scope Process -Force;[System.Net.ServicePointManager]::SecurityProtocol =[System.Net.ServicePointManager]::SecurityProtocol -bor 3072; iex((Новый объект System.Net.WebClient).DownloadString('https://community.chocolatey.org/install.ps1'))

• установить bazel (для сборки jaxlib)

choco install bazel

• установить msys2 (утилиты linux для bazel)

choco установить msys2

• постоянно связывать среду Python в вашей PowerShell

[System.Environment] :: SetEnvironmentVariable ("ПУТЬ", $Env:Path +";C:\path\ to \venv", "Machine")

• все еще находясь в папке jax в PowerShell, фактически создает и компилирует jax

./build/build.py

• после установки выполните команду, которую вы просите установить jaxlib, это что-то вроде

pip установить c:\...\jax\dist\jaxlib-0.1.72-cp39-none-win_amd64.whl

• и, наконец, установите с ним jax

pip install -e.

Таким образом, у меня это сработало, поскольку документация на домашней странице jax меня совершенно смутила.

Answer 4

В настоящее время JAX не предоставляет сборки jaxlib для Windows.

Выпуск 1, выпуск 2

Но вы можете построить его самостоятельно, если хотите. В этом выпуске есть несколько комментариев, которые могут вам помочь.

Answer 5

Это сработало для меня:

      powershell -ExecutionPolicy ByPass -NoExit -Command "& 'C:\users\<username>\Anaconda3\condabin\conda_hook.bat' ; conda activate <yourcondaEnvironment>"