当前位置:首页 > 系统教程 > 正文

Ubuntu 24极简安装PyTorch全流程(小白也能轻松上手的深度学习环境配置指南)

Ubuntu 24极简安装PyTorch全流程(小白也能轻松上手的深度学习环境配置指南)

欢迎阅读本教程!本文将带你一步步在Ubuntu 24系统上使用最简方式安装PyTorch,无论你是深度学习初学者还是经验丰富的开发者,都能快速完成环境配置。这也是全网最详细的PyTorch安装教程之一。

准备工作:Linux深度学习环境配置

首先,更新系统包,安装Python3和pip(如果未安装)。在终端执行以下命令:

    sudo apt updatesudo apt upgrade -ysudo apt install python3 python3-pip -y  

这里使用了Python包管理工具pip,后续安装PyTorch也会用到。

创建虚拟环境(可选但推荐)

为了避免包冲突,建议创建Python虚拟环境:

    sudo apt install python3-venv -ypython3 -m venv pytorch_envsource pytorch_env/bin/activate  

安装PyTorch

这是核心步骤。根据你的需求选择安装命令。如果你只需要CPU版本:

    pip install torch torchvision torchaudio  

如果需要CUDA支持(例如CUDA 11.8),使用:

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

等待安装完成。整个过程非常简单,遵循这个PyTorch安装教程,你很快就能完成。

验证安装

安装完成后,运行以下Python命令验证:

    python -c "import torch; print(torch.version); print(torch.cuda.is_available())"  

如果输出类似“2.0.1”和“True/False”,表示安装成功。下面是成功安装的截图示例:

Ubuntu 24极简安装PyTorch全流程(小白也能轻松上手的深度学习环境配置指南) 24 安装 PyTorch  安装教程 Linux 深度学习环境配置 Python 包管理工具 pip 第1张

常见问题

  • 下载速度慢:可以临时使用国内镜像源,如:pip install torch -i https://pypi.tuna.tsinghua.edu.cn/simple
  • 权限问题:在虚拟环境中安装可避免,或使用--user参数。
  • CUDA版本不匹配:请先确认你的CUDA版本,选择对应的PyTorch版本。

至此,你已经成功在Ubuntu 24上完成了PyTorch的安装!现在可以开始你的深度学习之旅了。

本教程关键词:Ubuntu 24 安装 PyTorch, PyTorch 安装教程, Linux 深度学习环境配置, Python 包管理工具 pip