Ubuntu に PyTorch と JupyterLab をインストールする

Ubuntu で PyTorch と JupyterLab をセットアップしましょう!このガイドでは、必要なものすべてをインストールして構成するための合理的なアプローチを提供します。

まず、Ubuntu システムが最新であることを確認します。

sudo apt update && sudo apt -y upgrade

Python のバージョンを確認してください。 次のステップで必要になるため、バージョン番号 (例: 3.13.x) をメモしておきます。

python3 --version

Python のバージョンに合わせて、適切な python3.xx-venv パッケージをインストールします。 Python のバージョンが 3.13.x の場合は、以下の 3.12 を 3.13 に置き換えてください。

sudo apt install -y python3.12-venv

仮想環境を作成します (これにより、プロジェクトの依存関係が分離されます):

python3 -m venv venv

仮想環境をアクティブ化します:

source venv/bin/activate

CUDA 11.8 をサポートする PyTorch をインストールします。 (別の CUDA バージョンまたは CPU のみのビルドを使用している場合は、必ず cu118 を調整してください。オプションについては、PyTorch Web サイトを参照してください。)。

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

JupyterLab をインストールします:

pip install jupyterlab

JupyterLab を起動し、必要に応じて作業ディレクトリを指定します。 /home/kai を目的のディレクトリに置き換えます。

jupyter lab --notebook-dir /home/kai

準備は完了です! 好みの Web ブラウザ (Firefox など) を開き、JupyterLab がターミナルで提供する URL に移動して作業を開始します。 deactivate の使用が終了したら、忘れずに仮想環境を非アクティブ化してください。

以上がUbuntu に PyTorch と JupyterLab をインストールするの詳細内容です。詳細については、PHP 中国語 Web サイトの他の関連記事を参照してください。