【Ubuntu2404】Ubuntu24.04安装PyTorch (GPU)
Ubuntu24.04安装PyTorch
- 引言
- 一、查看本机支持的CUDA版本
- 二、访问PyTorch官网进行下载
- 三、克隆base环境（可选）
- 四、安装Pytoch
- 五、检验是否可用
- 5.1 查看PyTorch是否可用
- 5.2 查看CUDA是否可用
- 5.3 查看cuDNN是否可用
引言
PyTorch 是由 Facebook 的人工智能研究实验室（FAIR）开发的一个开源机器学习库，它在深度学习领域非常受欢迎。PyTorch 提供了灵活的张量计算能力，类似于 NumPy，但它还支持 GPU 加速和自动求导功能，这使得它非常适合进行高效的数值运算，特别是针对神经网络的训练。本文主要介绍 Ubuntu24.04 版本下 PyTorch 的安装。
一、查看本机支持的CUDA版本
在终端中输入如下命令查看 CUDA 版本。
nvidia-smi
如表中右上角所示， CUDA 版本为 12.4 。
CUDA Version: 12.4
二、访问PyTorch官网进行下载
访问 PyTorch 官网查看本机版本的下载方式。
下载链接：PyTorch (官网)
页面往下翻可以看到如下选择安装环境，例如本机：
操作系统：Linux (Ubuntu 24.04)
包管理器：Conda
语言：Python
CUDA版本：12.4
其安装命令为：
conda install pytorch torchvision torchaudio pytorch-cuda=12.4 -c pytorch -c nvidia
三、克隆base环境（可选）
由于我使用的是 Anaconda3 ，使用时我不想影响 base 环境，但 base 环境下有很多已经安装好的包，所以考虑克隆 base 环境后安装 PyTorch。
打开终端，运行如下命令克隆 base 环境，新环境并命名为 pytoch_env：
conda create -n pytoch_env --clone base
克隆完成后查看并选择新环境：
conda env list
conda activate pytoch_env
四、安装Pytoch
使用命令，在新建的 pytoch_env 环境下安装 PyTorch：
conda install pytorch torchvision torchaudio pytorch-cuda=12.4 -c pytorch -c nvidia
安装过程中需要输入Y/y同意，或加上-y参数默认同意。
conda install -y pytorch torchvision torchaudio pytorch-cuda=12.4 -c pytorch -c nvidia
安装过程中需要下载的内容比较大，共计 1.6G 左右，但下载速度还可以，请耐心等待。
看到如下内容说明安装成功：
Preparing transaction: done
Verifying transaction: done
Executing transaction: done
五、检验是否可用
打开终端，并启用 Python ：
python
5.1 查看PyTorch是否可用
在 Python 环境中输入如下程序：
import torch
torch.__version__
如返回 PyTorch 版本信息，说明PyToch安装成功。
5.2 查看CUDA是否可用
在 Python 环境中输入如下程序：
import torch
torch.cuda.is_available()
如返回 True 则说明CUDA能够使用。
5.3 查看cuDNN是否可用
在 Python 环境中输入如下程序：
from torch.backends import cudnn
cudnn.is_available()
如返回 True 则说明cuDNN能够使用。
