pytorch的安装指南

• PyTorch 官网
• PyTorch GitHub

1. 安装前准备

首先，确保你已安装 Python 和 pip。如果没有安装，先安装 Python。你可以通过以下命令检查是否已安装：

python --version
pip --version

如果未安装，请访问 Python 官方网站 下载并安装最新版本的 Python。安装过程中勾选 Add Python to PATH 选项。

2. 选择安装方式

PyTorch 提供了多种安装方式。最常用的方式是通过 pip 或 conda 安装。你可以访问 PyTorch 官网 来根据你的系统和需求生成对应的安装命令。

1. 使用 pip 安装稳定版 PyTorch（支持 CPU）

pip install torch torchvision torchaudio

2. 安装 PyTorch（支持 GPU）

若要安装支持 CUDA 的 PyTorch（即 GPU 版本），你需要选择适合你的 CUDA 版本的 PyTorch 版本。
例如，如果你有 CUDA 11.8，可以使用以下命令：

pip install torch torchvision torchaudio cudatoolkit=11.8

3. 使用 Conda 安装（可选）

如果你使用的是 Anaconda，推荐通过 Conda 安装，因为它能自动处理依赖关系。

1. 安装支持 GPU 的版本(推荐)

conda install pytorch torchvision torchaudio cudatoolkit=11.8 -c pytorch

2. 安装稳定版 PyTorch（支持 CPU）

conda install pytorch torchvision torchaudio cpuonly -c pytorch

4. 验证安装

安装完成后，验证 PyTorch 是否安装成功：

import torch
print(torch.__version__)  # 打印 PyTorch 版本
print(torch.cuda.is_available())  # 检查是否能够使用 CUDA (GPU)

如果 torch.cuda.is_available() 返回 True，说明 PyTorch 可以使用 GPU。

5. 安装其他依赖

如果你需要额外的依赖，比如安装用于数据处理的 torchvision 或音频处理的 torchaudio，可以通过以下命令单独安装：

pip install torchvision torchaudio

6. 常见问题

CUDA 版本不兼容：确保你的显卡支持相应的 CUDA 版本，并且你已正确安装显卡驱动程序。 安装错误：如果安装过程中出现错误，建议先确保 Python 和 pip 版本都是最新的，并且已经安装了合适的依赖项。

参考资料
来自ai