Ubuntu 24.04 安装 PyTorch 详细教程（GPU版本与环境配置全攻略）

随着 Ubuntu 24.04 LTS 的发布，许多人工智能开发者开始在这一新平台上搭建科研环境。本文将为您详细介绍如何实现 Ubuntu 24.04 PyTorch 安装，即使是刚接触 Linux 的新手也能轻松上手。

第一步：安装 NVIDIA 驱动以实现 CUDA 深度学习加速

要运行 GPU 版本的 PyTorch，首先需要确保显卡驱动正确安装。Ubuntu 24.04 提供了非常简便的安装方式：

    sudo ubuntu-drivers autoinstall  

安装完成后重启电脑，使用 nvidia-smi 命令确认驱动是否生效。这是确保 CUDA 深度学习加速 能够正常工作的核心前提。

第二步：使用 Conda 进行 Python 环境配置

为了避免系统级 Python 依赖冲突，强烈建议使用 Miniconda 进行 Python 环境配置。创建一个专门的虚拟环境：

    conda create -n ai_env python=3.11conda activate ai_env  

第三步：正式执行 Ubuntu 24.04 PyTorch 安装

在激活的虚拟环境中，根据你的 CUDA 版本执行安装命令。对于绝大多数现代 NVIDIA 显卡，推荐安装 PyTorch GPU 版本：

    pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121  

图：PyTorch 环境安装流程参考

第四步：验证安装结果

最后一步是验证安装是否成功。请在终端输入 python，然后输入以下代码：

    import torchprint(f'PyTorch 版本: {torch.__version__}')print(f'是否支持GPU: {torch.cuda.is_available()}')  

如果输出为 True，恭喜你，Ubuntu 24.04 PyTorch 安装 已经圆满完成！现在你可以开启你的深度学习之旅了。