本文解決圖書《谷歌JAX深度學(xué)習(xí)從零開始學(xué)》的環(huán)境搭建問(wèn)題。谷歌JAX深度學(xué)習(xí)框架要求運(yùn)行在Linux系統(tǒng)上，我們可以在Windows11系統(tǒng)中，利用WSL虛擬環(huán)境搭建Ubuntu虛擬機(jī)系統(tǒng)，在Windows11上運(yùn)行PyCharm集成開發(fā)環(huán)境，調(diào)用Ubuntu虛擬機(jī)系統(tǒng)上的JAX框架，對(duì)代碼進(jìn)行編程和調(diào)試。（1）首先，安裝Windows Subsystem for Linux（WSL）。在Windows11設(shè)置窗口上，搜索出“啟用或關(guān)閉Windows功能”選項(xiàng)，并勾選“適用于Linux的Windows子系統(tǒng)”選項(xiàng)。重啟計(jì)算機(jī)。

（2）第二，啟用開發(fā)者模式。在“Windows設(shè)置”中搜索“開發(fā)人員”，打開“開發(fā)者選項(xiàng)”窗口，把開發(fā)人員模式打開。

（3）第三，升級(jí)WSL系統(tǒng)。

在Windows終端中以管理員身份運(yùn)行如下命令：

wsl.exe --update

等待升級(jí)結(jié)束后運(yùn)行如下命令：

wsl --set-default-version 2

可以通過(guò)如下命令查看WSL的版本號(hào)：

wsl --list –verbose

（4）第四，從“Microsoft Store”中安裝Ubuntu。打開Microsoft Store頁(yè)面，搜索Ubuntu，在搜索的結(jié)果中選擇安裝Ubuntu 20.04版本的Linux虛擬機(jī)。Microsoft Store頁(yè)面打不開，自己百度一下解決。

單擊“get”按鈕即開始安裝，安裝過(guò)程很簡(jiǎn)單。

500多M很快就裝完了，設(shè)置用戶名密碼即可。菜單上會(huì)出來(lái)一個(gè)圖標(biāo)“Ubuntu 20.04.5 LTS”，點(diǎn)擊圖標(biāo)即可打開Ubuntu終端。

（5）第五，新安裝的WSL需要更新一次，打開WSL終端界面，依次輸入如下操作語(yǔ)句：

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的安裝和驗(yàn)證：

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

這是啟動(dòng)WSL自帶的Python命令，之后鍵入如下命令：

import jax.numpy as np

np.add(1.0,1.7)

如下結(jié)果說(shuō)明安裝無(wú)誤。

書中PyCharm用了社區(qū)版，配置起來(lái)非常麻煩。（7）第七，請(qǐng)讀者在Windows11上直接使用PyCharm專業(yè)版，方便在編譯器上直接配置WSL虛擬環(huán)境。

注意：JAX相關(guān)包的安裝都在Ubuntu虛擬機(jī)系統(tǒng)上做的，源碼中缺什么包就安裝什么包。