news 2026/4/18 8:27:03

在Windows系统上安装PyTorch GPU版：从零到成功运行的过程记录

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
在Windows系统上安装PyTorch GPU版：从零到成功运行的过程记录

作为深度学习研究者或开发者，能够利用GPU加速模型训练是提高工作效率的关键。本文将详细介绍如何在Windows系统上成功安装并配置PyTorch GPU版本，让你能够充分利用NVIDIA显卡的计算能力。

第一步：检查当前系统环境

在开始安装之前，我们首先需要了解当前的硬件和驱动情况。

1.1 查看GPU信息

打开命令提示符（CMD）或PowerShell，输入以下命令：

nvidia-smi

根据你的输出，我们可以看到：

+-----------------------------------------------------------------------------------------+ | NVIDIA-SMI 551.86 Driver Version: 551.86 CUDA Version: 12.4 | |-----------------------------------------+------------------------+----------------------+ | GPU Name TCC/WDDM | Bus-Id Disp.A | Volatile Uncorr. ECC | | Fan Temp Perf Pwr:Usage/Cap | Memory-Usage | GPU-Util Compute M. | | | | MIG M. | |=========================================+========================+======================| | 0 NVIDIA GeForce RTX 3050 WDDM | 00000000:01:00.0 On | N/A | | 31% 30C P8 4W / 70W | 1337MiB / 6144MiB | 6% Default | +-----------------------------------------+------------------------+----------------------+

关键信息解读：

  • 驱动版本: 551.86
  • CUDA版本: 12.4
  • GPU型号: NVIDIA GeForce RTX 3050
  • 显存: 6GB

1.2 检查Python版本

确保你已经安装了Python 3.8（或其他兼容版本）：

python --version

第二步：安装CUDA Toolkit

2.1 下载合适的CUDA版本

根据nvidia-smi显示的CUDA版本（12.4），我们需要安装对应的CUDA Toolkit：

  1. 访问NVIDIA CUDA Toolkit Archive
  2. 选择CUDA 12.4版本（与驱动兼容）
  3. 选择Windows系统对应的安装包

2.2 安装CUDA Toolkit

  1. 运行下载的安装程序
  2. 选择"自定义安装"
  3. 建议保持默认选项，但可以取消Visual Studio集成（如果你不需要）
  4. 完成安装后，验证安装：
nvcc --version

第三步：安装cuDNN

3.1 下载cuDNN

  1. 访问cuDNN Archive
  2. 选择与CUDA 12.4兼容的cuDNN版本
  3. 需要注册NVIDIA账户（免费）

3.2 安装cuDNN

  1. 下载ZIP文件后解压
  2. 将解压后的文件复制到CUDA安装目录：
    • 将bin文件夹内容复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.4\bin
    • 将include文件夹内容复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.4\include
    • 将lib文件夹内容复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.4\lib

第四步：安装PyTorch GPU版本

4.1 选择合适的PyTorch版本

由于我们使用的是CUDA 12.1/12.4，建议使用对应的PyTorch版本。

4.2 使用pip安装

打开命令提示符或PowerShell，运行以下命令：

pipinstalltorch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121 --trusted-host download.pytorch.org

参数说明：

  • --index-url: 指定PyTorch的CUDA 12.1版本仓库
  • --trusted-host: 信任该主机，避免SSL验证问题

注意：虽然nvidia-smi显示CUDA 12.4，但PyTorch官方提供的是CUDA 12.1版本，这两个版本通常是兼容的。

4.3 替代安装方法

如果你遇到网络问题，可以使用清华镜像源：

pipinstalltorch torchvision torchaudio -i https://pypi.tuna.tsinghua.edu.cn/simple

第五步：验证安装

5.1 创建测试脚本

创建一个Python文件（如test_gpu.py），包含以下代码：

importtorchprint("=== PyTorch GPU检测 ===")print(f"PyTorch版本:{torch.__version__}")print(f"CUDA是否可用:{torch.cuda.is_available()}")print(f"CUDA版本:{torch.version.cuda}")iftorch.cuda.is_available():print(f"GPU数量:{torch.cuda.device_count()}")foriinrange(torch.cuda.device_count()):print(f"\n--- GPU{i}---")print(f"名称:{torch.cuda.get_device_name(i)}")print(f"内存总量:{torch.cuda.get_device_properties(i).total_memory/1e9:.2f}GB")print(f"当前内存使用:{torch.cuda.memory_allocated(i)/1e9:.2f}GB")print(f"最大内存使用:{torch.cuda.max_memory_allocated(i)/1e9:.2f}GB")# 简单张量计算测试print("\n=== GPU计算测试 ===")device=torch.device("cuda"iftorch.cuda.is_available()else"cpu")x=torch.randn(10000,10000).to(device)y=torch.randn(10000,10000).to(device)z=torch.matmul(x,y)print(f"矩阵计算完成，结果形状:{z.shape}")print("✅ GPU加速工作正常！")else:print("\n❌ CUDA不可用！可能的原因：")print("1. 没有安装CUDA版本的PyTorch")print("2. NVIDIA驱动太旧")print("3. PyTorch与CUDA版本不匹配")print("4. CUDA/cuDNN没有正确安装")print("5. 系统PATH环境变量未正确配置")

5.2 运行测试

python test_gpu.py

5.3 预期成功输出

=== PyTorch GPU检测 === PyTorch版本: 2.3.0+cu121 CUDA是否可用: True CUDA版本: 12.1 --- GPU 0 --- 名称: NVIDIA GeForce RTX 3050 内存总量: 6.00 GB 当前内存使用: 0.00 GB 最大内存使用: 0.00 GB === GPU计算测试 === 矩阵计算完成，结果形状: torch.Size([10000, 10000]) ✅ GPU加速工作正常！

第六步：常见问题解决

6.1 CUDA不可用

如果torch.cuda.is_available()返回False：

  1. 检查驱动版本：确保NVIDIA驱动是最新的
  2. 验证CUDA安装：运行nvcc --version确认CUDA已安装
  3. 检查环境变量：确保CUDA路径已添加到系统PATH
  4. 重新安装PyTorch：使用正确的CUDA版本

6.2 安装速度慢

  • 使用国内镜像源
  • 使用conda安装（如果需要）：
condainstallpytorch torchvision torchaudio pytorch-cuda=12.1-c pytorch -c nvidia

6.3 版本冲突

如果遇到版本冲突，建议创建虚拟环境：

# 创建虚拟环境python -m venv pytorch_gpu# 激活虚拟环境（Windows）pytorch_gpu\Scripts\activate# 在虚拟环境中安装pipinstalltorch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

第七步：性能优化建议

7.1 设置GPU内存分配策略

# 在代码开头添加torch.cuda.empty_cache()torch.backends.cudnn.benchmark=True# 自动优化卷积算法

7.2 监控GPU使用情况

使用以下命令实时监控GPU状态：

# 每隔1秒刷新一次nvidia-smi -l1

总结

通过以上步骤，你应该已经成功在Windows系统上安装了PyTorch GPU版本。关键要点：

  1. 版本匹配至关重要：确保驱动、CUDA、cuDNN和PyTorch版本兼容
  2. 按顺序安装：驱动 → CUDA → cuDNN → PyTorch
  3. 验证每一步：使用命令行工具验证每个组件的安装
  4. 利用虚拟环境：避免包冲突，保持环境清洁

现在你可以开始使用GPU加速的PyTorch进行深度学习项目开发了！如果遇到任何问题，欢迎在评论区留言讨论。

相关资源链接：

  • PyTorch官方安装指南
  • NVIDIA驱动下载
  • CUDA Toolkit Archive
  • cuDNN Archive

至此记录完毕，respect！

版权声明: 本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如若内容造成侵权/违法违规/事实不符，请联系邮箱：809451989@qq.com进行投诉反馈，一经查实，立即删除！
网站建设 2026/4/17 19:00:31

【MYSQL数据库】基础版之表的操作

3.表的操作3.1 创建表语法：create table table_name(filed1 datatype,filed2 datatype,filed3 datatype )character set 字符集 collate 校验规则 engine 存储引擎;说明：field 表示列名datatype 表示列的类型character set 字符集，如果没有指…

作者头像 李华
网站建设 2026/4/18 7:58:06

提示词工程解锁大模型潜力的终极沟通术

提示词工程是提升AI交互效果的关键技术，主要包含User Prompt和System Prompt两类。User Prompt是用户即时指令，需遵循清晰表达、角色设定和示例提供等原则；System Prompt则是全局性指令集，定义AI角色和行为准则。通过RTF框架结构化…

作者头像 李华
网站建设 2026/4/18 7:28:02

仅限内部分享：资深专家解读Dify重排序日志的7种模式

第一章：检索重排序的 Dify 日志分析在构建基于大语言模型的应用时，Dify 作为低代码开发平台，提供了完整的日志追踪机制，尤其在检索增强生成（RAG）流程中，对检索与重排序阶段的日志分析至关重要。…

作者头像 李华
网站建设 2026/4/11 20:26:57

手把手教你实现Dify模型秒级加载，私有化部署不再卡顿

第一章：私有化部署的 Dify 模型加载在企业级 AI 应用场景中，私有化部署 Dify 成为保障数据安全与合规性的首选方案。通过本地化运行 Dify，企业可在内网环境中完成大模型的接入、编排与服务发布，避免敏感数据外泄。模型加载作为核心…

作者头像 李华
网站建设 2026/4/17 7:35:25

好写作AI｜别让AI只改病句了！它已潜入你论文的“思想后台”

当你还让AI检查“的得地”，学霸已用它来辩论学术观点了各位在学术深水区“扑腾”的研究者，是时候更新一下对AI写作工具的认知了！如果你的使用记录还停留在“帮我改改语病”和“这段降个重”，那可真有点“杀鸡用牛刀”了。今天&…

作者头像 李华
网站建设 2026/4/16 2:55:35

通俗易懂讲线程--适合小白的零基础教程（面试版）

作者有话要说： 本文章适合想要快速了解和学习线程的基本知识点，适合考试复习和准备面试的同学。 一，了解线程和进程 官方给出的关于这两个名词的具体概念解释如下： 进程（Process）是系统进行资源分配的…

作者头像 李华