无法安装jaxlib

Question

我正在尝试通过我在文档中找到的以下命令在我的 Windows 10 上安装 jaxlib 。

pip 安装 jaxlib

它显示以下错误

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

Answer 1

感谢cloudhan的jax-windows-builder ，现在在 Windows 上安装JAXjaxlib相当容易。例如

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

就这样。

正如那里所解释的，我必须复制我感兴趣的whl 文件的jaxlib链接地址，即上面的）。https://whls.blob.core...

但我还必须注意 JAX 的版本和 Jaxlib 的版本是兼容的，这种兼容性很容易在github.com/google/jax/releases上评估：只需选择每个（！）^版本号的最后两个！^{没有什么可从那里下载的。}

使用上面明确的版本进行了测试。_{即 python3.8-64 & jax==0.3.13 & jaxlib-0.3.7}

Answer 2

Windows 不支持 Jaxlib，您可以在此处查看它。 https://github.com/google/jax/issues/438

Answer 3

我自己在过去两天经历了这个过程，所以这就是我所做的：

1. 下载并安装最新版本的 microsoft Visual Studio（以获得 C++ 编译器和工具链）

2. 下载并安装python

3. 使用 pip、setuptools、six、wheel 和 numpy 包创建虚拟 python 环境（我在 pycharm 的 GUI 中完成）

4. 下载贾克斯

5. 以管理员身份打开 Windows powershell，切换到 jax 目录并完成以下步骤（命令在引号中）

• 安装chocolatey（用于轻松安装bazel的包管理器）

Set-ExecutionPolicy 绕过 -Scope Process -Force；[System.Net.ServicePointManager]::SecurityProtocol = [System.Net.ServicePointManager]::SecurityProtocol -bor 3072; iex ((New-Object System.Net.WebClient).DownloadString('https://community.chocolatey.org/install.ps1'))

• 安装 bazel （用于构建 jaxlib）

choco 安装 bazel

• 安装 msys2（bazel 的 Linux 实用程序）

choco安装msys2

• 永久链接 powershell 中的 python 环境

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

• 仍在 powershell 中的 jax 文件夹中实际构建和编译 jax

./build/build.py

• 安装后执行要求安装 jaxlib 的命令，就像这样

pip install c:\...\jax\dist\jaxlib-0.1.72-cp39-none-win_amd64.whl

• 最后安装jax

pip install -e 。

这种方式对我有用，因为 jax 主页上的文档让我完全困惑。