无法安装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

更新20240514

Jax 现在在 Windows 上受支持，因为 Jaxlib 自 2023 年 3 月以来一直如此。

---

感谢cloudhan的jax-windows-builder，现在在Windows上安装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

仅此而已。正如那里所解释的，我必须我感兴趣的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. 下载jax

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

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

Set-ExecutionPolicy 绕过-Scope Process-Force； [System.Net.ServicePointManager]::SecurityProtocol = [System.Net.ServicePointManager]::SecurityProtocol -bor 3072;埃克斯 ((新对象 System.Net.WebClient).DownloadString('https://community.chocolatey.org/install.ps1'))

• 安装 bazel（用于构建 jaxlib）

choco 安装 bazel

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

choco 安装 msys2

• 在你的powershell中永久链接python环境

[系统.环境]::SetEnvironmentVariable("PATH", $Env:Path + ";C:\env 路径", "机器")

• 仍在 powershell 中的 jax 文件夹中实际构建和编译 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 目前不提供适用于 Windows 的 jaxlib 构建。

第 1 期，第 2 期

但是如果您愿意，您可以自己构建它。上期中的一些评论可能会对您有所帮助。

Answer 5

这对我有用：

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