在Ubuntu上安装Pytorch和Jupyterlab

请我喝杯咖啡☕

第一步，更新您的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 download.pytorch.org/whl/cu118

安装JupyterLab：

pip install jupyterlab

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

jupyter lab

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

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

以上就是在Ubuntu上安装Pytorch和Jupyterlab的详细内容，更多请关注范的资源库其它相关文章！