在Ubuntu上安装Pytorch和Jupyterlab

百变鹏仔 4小时前 #Python

文章标签 Ubuntu

请我喝杯咖啡☕

第一步，更新您的Ubuntu系统：

sudo apt update && sudo apt -y upgrade

确认Python版本：

python3 --version# 例如：Python 3.12.3

根据您的Python版本安装venv： (例如，Python 3.13.x 需要安装 python3.13-venv) Ubuntu系统默认不包含此包。

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

查看当前目录：

pwd# /home/kai (示例)

创建一个虚拟环境：

python3 -m venv venv

激活虚拟环境：

source venv/bin/activate

使用CUDA 11.8安装PyTorch (您可以根据您的CUDA版本或使用CPU版本调整安装命令)：

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

安装JupyterLab：

pip install jupyterlab

启动JupyterLab (可在Firefox或其他浏览器访问)：

jupyter lab

或者，指定工作目录启动JupyterLab：

jupyter lab --notebook-dir /home/kai  # 请替换为您的工作目录

文章推荐