私は過去2日間自分でプロセスを行ったので、私がしたことは次のとおりです。
Microsoft Visual Studio の最新バージョンをダウンロードしてインストールします (C++ コンパイラとツールチェーンを入手するため)。
パイソンをダウンロードしてインストールする
パッケージpip、setuptools、six、wheel、およびnumpyを使用して仮想python環境を作成します(pycharmのGUIで行いました)
ジャックスをダウンロード
- 管理者として Windows PowerShell を開き、jax ディレクトリに移動して、次の手順を実行します (コマンドは引用符で囲まれています)。
- Chocolatey をインストールします (bazel を簡単にインストールするためのパッケージ マネージャー)
Set-ExecutionPolicy Bypass -Scope Process -Force; [System.Net.ServicePointManager]::SecurityProtocol = [System.Net.ServicePointManager]::SecurityProtocol -bor 3072; iex ((New-Object System.Net.WebClient).DownloadString('https://community.chocolatey.org/install.ps1'))
- bazel のインストール (jaxlib のビルド用)
チョコインストールバゼル
- msys2 (bazel 用の Linux ユーティリティ) をインストールします。
チョコインストール msys2
- PowerShell で Python 環境を永続的にリンクする
[System.Environment]::SetEnvironmentVariable("PATH", $Env:Path + ";C:\path\to\venv", "Machine")
- PowerShell の jax フォルダーにまだあり、実際に jax をビルドしてコンパイルします。
./build/build.py
- インストール後、jaxlib をインストールするように要求されたコマンドを実行します。これは次のようなものです。
pip install c:\...\jax\dist\jaxlib-0.1.72-cp39-none-win_amd64.whl
pip インストール -e 。
jaxホームページのドキュメントが私を完全に混乱させたので、この方法でうまくいきました。