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

brucexia發表於2022-10-17

谷歌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深度學習從零開始學》那本書開發環境。









來自 “ ITPUB部落格 ” ,連結:http://blog.itpub.net/18841117/viewspace-2918790/,如需轉載,請註明出處,否則將追究法律責任。

相關文章