在 Ubuntu 上安装 PyTorch 和 JupyterLab
谢谢您的咖啡☕!以下是经过润色后的文本,在保持原意的基础上,对语句进行了调整,并保留了图片的原始格式:
升级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
改进说明:
希望修改后的版本更符合您的要求。