摘要:本文手把手教你安装PyTorch深度学习框架,涵盖Anaconda/pip两种安装方式,详细讲解CPU与GPU版本选择、CUDA配置及环境验证,解决99%的安装报错问题!
一、PyTorch简介
PyTorch是Meta(原Facebook)开源的深度学习框架,以其动态计算图和易用性广受研究人员喜爱。支持GPU加速训练,是AI开发者的首选工具。
二、安装前准备
- 硬件检查:
- 确认显卡型号(NVIDIA显卡需支持CUDA)
- Windows用户安装NVIDIA驱动
- 查看CUDA版本(GPU用户):
nvidia-smi # 右上角显示最高支持的CUDA版本
- Python环境:需3.7+版本(建议通过Anaconda管理)
三、通过Anaconda安装(推荐)
▶ 步骤1:创建虚拟环境
conda create -n pytorch_env python=3.9
conda activate pytorch_env
▶ 步骤2:访问PyTorch官网获取安装命令
- 打开官网:https://pytorch.org/
- 选择配置:
- PyTorch版本:稳定版(Stable)
- 操作系统:根据实际情况选择
- 包管理器:Conda/Pip
- 语言:Python
- CUDA:选