跳转到内容

PyTorch安装指南

1. 简介

PyTorch 是由 Meta（前 Facebook）AI 研究团队开发的开源机器学习库。它广泛应用于计算机视觉和自然语言处理等深度学习领域，因其 动态计算图（Eager Execution）和与 Python 生态的紧密集成，深受科研人员与工程开发者的喜爱。

2. 安装

通常可以通过 Python 包管理器 pip 快速安装 PyTorch 的 CPU 及 GPU（CUDA）通用驱动版本：

pip install torch torchvision torchaudio

3. 验证 GPU 加速 (CUDA)

安装完毕后，可以通过以下 Python 代码在命令行或 Notebook 中检测 PyTorch 是否已成功识别到系统的 NVIDIA 显卡：

import torch

# 1. 检查 PyTorch 版本
print(f"PyTorch Version: {torch.__version__}")

# 2. 检查 CUDA (GPU) 是否可用
cuda_available = torch.cuda.is_available()
print(f"CUDA Available: {cuda_available}")

if cuda_available:
    # 3. 获取可用 GPU 的数量与名称
    print(f"GPU Count: {torch.cuda.device_count()}")
    print(f"Current GPU Name: {torch.cuda.get_device_name(0)}")

4. 参考资料