无法安装 jaxlib

Question

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

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 提供 jaxlib 构建。

Issue 1, issue 2

但如果您wish.There在上述问题中的一些评论可能对您有帮助，您可以自己构建它。

Answer 3

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

下载并安装最新版本的 Microsoft visual studio ( 获得 C++ 编译器和工具链）
下载并安装python
使用包 pip、setuptools、six、wheel 和 numpy 创建虚拟 python 环境（我是在 pycharm 的 GUI 中完成的）
download jax

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

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

Set-ExecutionPolicy Bypass -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 install bazel

安装 msys2（linux bazel 实用程序）

choco install msys2

永久地link你的 powershellpython 环境

[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 主页上的文档让我完全困惑。

Answer 4

这对我有用：

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

Answer 5

我可能错过了一些东西，但是，感谢 windows 上的 cloudhan's jax-windows-builder, it is now rather easy to install JAX and 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

正如我所解释的 there, I had to copy the jaxlib's link address of the whl file 我感兴趣的，即上面的 https://whls.blob.core...）。

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

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

无法安装 jaxlib

Not able to install jaxlib

python

installation

numpy

pip

jax