numpy 无法安装jaxlib

66bbxpm5  于 2023-04-30  发布在  其他
关注(0)|答案(6)|浏览(319)

我试图通过以下命令在我的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
k97glaaz

k97glaaz1#

多亏了cloudhanjax-windows-builder,现在在windows上安装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

就这样
正如这里所解释的,我不得不使用copy,我感兴趣的whl filejaxlib 的链接地址i。即上面的https://whls.blob.core...)。
但我还必须注意JAX的版本和Jaxlib的版本是兼容的,在github.com/google/jax/releases上很容易评估兼容性:只要选择每个的最后两个 *(!)*版本号!没有什么可以下载的。

测试使用上述版本 即python3。8-64&jax==0.3.13&**jaxlib-0.3.7

xienkqul

xienkqul2#

Jaxlib不支持Windows,你可以在这里看到它。. https://github.com/google/jax/issues/438

krcsximq

krcsximq3#

我在过去的两天里经历了这个过程,所以我做了如下:
1.下载并安装最新版本的microsoft visual studio(以获得c++编译器和工具链)
1.下载并安装Python
1.使用pip、setuptools、six、wheel和numpy包创建一个虚拟python环境(我在pycharm的GUI中完成了)

  1. download jax
    1.打开一个windows powershell作为管理员,改变到jax目录并完成以下步骤(命令在引号中)
  • 安装chocolatey(用于轻松安装bazel的包管理器)

Set-ExecutionPolicy Bypass -Scope Process -Force;[System.Net.ServicePointManager]::SecurityProtocol = [System.Net.ServicePointManager]::SecurityProtocol -bor 3072;iex((新对象系统。WebClient)。DownloadString('https://community.chocolatey.org/install.ps1'))

  • 安装bazel(用于构建jaxlib)

choco install bazel

  • 安装msys 2(linux utilities for bazel)

choco安装msys 2

  • 永久性地将python环境链接到您的powershell中

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

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

./build/ www.example.com

  • 安装后执行命令你被要求安装jaxlib,它的东西像

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

  • 最后安装jax

pip install -e .
这种方式对我很有效,因为jax主页上的文档完全把我搞糊涂了。

bhmjp9jg

bhmjp9jg4#

你可以访问这个网站 www.example.com 会发现安装说明.

wkftcu5l

wkftcu5l5#

JAX目前还没有为Windows提供jaxlib版本。
Issue 1issue 2
但如果你愿意,你可以自己建造。上面的一些评论可能会对你有所帮助。

mm9b1k5b

mm9b1k5b6#

这对我很有效:

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

相关问题