【PyTorch】2024保姆级安装教程-Python-（CPU+GPU详细完整版）

【PyTorch】2024保姆级安装教程（CPU+GPU详细完整版）

PyTorch是当前最受欢迎的深度学习框架之一。本文将详细讲解在Python环境中安装PyTorch，包括CPU和GPU版本的全方位指南。

一、前置环境

首先确保已安装Python环境，推荐使用Python 3.8或以上版本。

验证Python安装：

python --version
pip --version

推荐使用虚拟环境（如conda或venv）：

使用conda创建虚拟环境（推荐）

conda create -n pytorch_env python=3.10
conda activate pytorch_env

使用venv创建虚拟环境

python -m venv pytorch_env
source pytorch_env/bin/activate  # Linux/macOS
pytorch_env\Scripts\activate.bat  # Windows

二、安装PyTorch CPU版本

适合没有NVIDIA显卡或不使用GPU的用户。

安装命令：

pip install torch torchvision torchaudio
<