在Ubuntu上安装Pytorch和Jupyterlab

请我喝杯咖啡☕

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

sudo apt update && sudo apt -y upgrade

登录后复制

确认Python版本：

python3 --version # 例如：Python 3.12.3

登录后复制

根据您的Python版本安装venv： (例如，Python 3.13.x 需要安装 3.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  # 请替换为您的工作目录

登录后复制

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

四平甲倪网络网站制作专家