在 Ubuntu 上安裝 PyTorch 和 JupyterLab

讓我們在 Ubuntu 上設定 PyTorch 和 JupyterLab！本指南提供了一種簡化的方法來安裝和配置您所需的一切。

首先，確保您的 Ubuntu 系統是最新的：

<code class="language-bash">sudo apt update && sudo apt -y upgrade</code>

檢查您的 Python 版本。 請記下版本號碼（例如 3.13.x），因為下一步將需要它。

<code class="language-bash">python3 --version</code>

安裝與您的 Python 版本相符的適當的 python3.xx-venv 套件。 如果您的 Python 版本是 3.13.x，請將下面的 3.12 替換為 3.13.

<code class="language-bash">sudo apt install -y python3.12-venv</code>

建立虛擬環境（這使您的專案依賴項保持獨立）：

<code class="language-bash">python3 -m venv venv</code>

啟動虛擬環境：

<code class="language-bash">source venv/bin/activate</code>

安裝支援 CUDA 11.8 的 PyTorch。 （如果您使用不同的 CUDA 版本或僅 CPU 版本，請記住調整 cu118；請參閱 PyTorch 網站以了解選項）。

<code class="language-bash">pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118</code>

安裝 JupyterLab：

<code class="language-bash">pip install jupyterlab</code>

啟動 JupyterLab，可以選擇指定您的工作目錄。 將 /home/kai 替換為您想要的目錄。

<code class="language-bash">jupyter lab --notebook-dir /home/kai</code>

一切都準備好了！ 打開您喜歡的網頁瀏覽器（例如​​ Firefox）並導航到終端機中提供的 URL JupyterLab 開始工作。 使用完deactivate後請記得停用虛擬環境。

以上是在 Ubuntu 上安裝 PyTorch 和 JupyterLab的詳細內容。更多資訊請關注PHP中文網其他相關文章！