【PyTorch】深度学习框架 - 搭建环境

Pytorch 介绍

PyTorch 是一个功能强大且灵活的深度学习框架，特别适合研究和开发。它的动态计算图、Pythonic 设计和丰富的工具使其成为深度学习领域的首选工具之一。无论是初学者还是资深开发者，都可以通过 PyTorch 快速实现和部署深度学习模型，本篇介绍如何在windows上搭建Pytorch开发环境。

---

---

前言

Pytorch需要用到线性代数，优化理论，机器学习，深度学习等理论，所以在学习Pytorch过程中需要补充相关知识。

---

一、环境搭建

注意：由于Pytroch需要用到GPU运算，所以安装推荐用双系统（带Linux）或直接用windows，不建议用虚拟机，本篇在windows上安装环境

1. 安装显卡驱动

通过设备管理器查看显卡驱动，如果驱动没装好，需要去官网装上（没装上驱动会有小黄标）。

2. 安装CUDA

安装CUDA是为了利用GPU加速运算，打开显卡面板检查显卡支持的CUDA版本，我使用的是CUDA12.6.1，