谷歌JAX深度學習開發環境搭建

谷歌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 jax==0.2.19  

pip install -i 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 虛擬機器系統上做的，原始碼中缺什麼包就安裝什麼包。

好了。谷歌JAX框架的深度學習程式碼可以執行起來了。

注意：本文可以用於搭建《谷歌JAX深度學習從零開始學》那本書開發環境。