在 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 网站了解选项）。

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

一切都准备好了！ 打开您喜欢的网络浏览器（例如 Firefox）并导航到终端中提供的 URL JupyterLab 开始工作。 使用完deactivate后请记得停用虚拟环境。

以上是在 Ubuntu 上安装 PyTorch 和 JupyterLab的详细内容。更多信息请关注PHP中文网其他相关文章！