环境：
Ubuntu 20.04
jax-0.4.17（进行源码编译）
Jaxlib 0.4.16（直接pip）
numpy>=1.22 （实际1.26）
scipy>=1.7
Python（3.9 3.10 3.11）实际选择3.9
（依赖版本jax/setup.py at jaxlib-v0.4.16 · google/jax · GitHub）
官方手册
Building from source — JAX documentation
网址：
jaxlib
GitHub - google/jax at jaxlib-v0.4.16
jax
开始：
1.安装python环境、依赖
python 3.9、numpy>=1.22、wheel、build其他未注明版本的直接pip安装即可
2.安装jaxlib
可以通过源码安装，也可以直接pip
先尝试源码安装：
在jaxlib根目录下执行
失败！！
很多原因，such as go语言下载不成功、build文件中输出未指定等。（先跳过直接pip安装，此时未仔细查看原因）
简单点选择直接pip下载jaxlib
Sudo python3.9 -m pip install jaxlib==0.4.16（此处有多个python，指定用python3.9的pip）
3.引入jax源码，然后install jax
在根目录下执行下面的命令
测试：
成功安装jax！！
扫一扫
1万+
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
