PHP前端开发

在 Ubuntu 上安装 PyTorch 和 JupyterLab

百变鹏仔 6天前 #Python
文章标签 Ubuntu

谢谢您的咖啡☕!以下是经过润色后的文本,在保持原意的基础上,对语句进行了调整,并保留了图片的原始格式:

升级Ubuntu系统并配置Python环境:

首先,更新您的Ubuntu系统:

sudo apt update && sudo apt -y upgrade

确认Python版本:

python3 --version# 例如:python 3.12.3

安装虚拟环境工具:

请根据您的Python版本安装相应的虚拟环境工具。例如,如果您使用的是Python 3.13.x,则需要安装python3.13-venv。Ubuntu系统默认不包含此工具。

sudo apt install -y python3.12-venv  # 请替换为您的Python版本

当前目录确认:

pwd# /home/kai

创建虚拟环境:

python3 -m venv venv

激活虚拟环境:

source venv/bin/activate

安装PyTorch (CUDA 11.8):

您可以根据您的硬件环境选择合适的PyTorch版本(CUDA、ROCm或CPU)。以下命令安装支持CUDA 11.8的PyTorch:

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

安装JupyterLab:

pip install jupyterlab

启动JupyterLab:

在Firefox浏览器中打开JupyterLab:

jupyter lab

您也可以指定工作目录:

jupyter lab --notebook-dir /home/kai

改进说明:

希望修改后的版本更符合您的要求。