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

原文由 user13782377 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 1

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

原文由 user13815693 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 2

感谢 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

原文由 keepAlive 发布，翻译遵循 CC BY-SA 4.0 许可协议