谷歌JAX 深度学习框架要求运行在Linux 系统上，我们可以在Windows11 系统中，利用WSL 虚拟环境搭建Ubuntu 虚拟机系统，在Windows11 上运行PyCharm 集成开发环境，调用Ubuntu 虚拟机系统上的JAX 框架，对代码进行编程和调试。
（1 ）首先，安装Windows Subsystem for Linux （WSL ）。在Windows11 设置窗口上，搜索出“启用或关闭Windows 功能”选项，并勾选“适用于Linux 的Windows 子系统”选项。重启计算机。
（2 ）第二，启用开发者模式。在“Windows 设置”中搜索“开发人员”，打开“开发者选项”窗口，把开发人员模式打开。
（3 ）第三，升级WSL 系统。
在Windows 终端中以管理员身份运行如下命令：
wsl.exe --update
等待升级结束后运行如下命令：
wsl --set-default-version 2
可以通过如下命令查看WSL 的版本号：
wsl --list –verbose
（4 ）第四，从“Microsoft Store ”中安装Ubuntu 。打开Microsoft Store 页面，搜索Ubuntu ，在搜索的结果中选择安装Ubuntu 20.04 版本的Linux 虚拟机。Microsoft Store 页面打不开，自己百度一下解决。
单击“get ”按钮即开始安装，安装过程很简单。
500 多M 很快就装完了，设置用户名密码即可。菜单上会出来一个图标 “Ubuntu 20.04.5 LTS ” ，点击图标即可打开Ubuntu 终端。
（5 ）第五，新安装的WSL 需要更新一次，打开WSL 终端界面，依次输入如下操作语句：
sudo apt update
sudo apt install gcc make g++
sudo apt install build-essential
sudo apt install python3-pip
pip install --upgrade pip
（6 ）第六，JAX 的安装和验证：
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple jax==0.2.19
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple jaxlib==0.1.70
python3
这是启动WSL 自带的Python 命令，之后键入如下命令：
import jax.numpy as np
np.add(1.0,1.7)
如下结果说明安装无误。
书中PyCharm 用了社区版，配置起来非常麻烦。
（7 ）第七，请读者在Windows11 上直接使用PyCharm 专业版，方便在编译器上直接配置WSL 虚拟环境。
注意：JAX 相关包的安装都在Ubuntu 虚拟机系统上做的，源码中缺什么包就安装什么包。
