有关PyTorch的全面介绍，可以参考我的另外一篇文章：https://blog.csdn.net/weixin_50366304/article/details/145521705?spm=1001.2014.3001.5502

1、PyTorch的安装

1. 准备工作

2. pytorch需要python3.6及以上的python版本

3. 我是利用Anaconda来管理我的python，安装conda，可以参考这篇文章，非常详细：BG大龍：【Anaconda教程01】怎么安装Anaconda3

4. Anaconda官网 Free Download | Anaconda

二、Pyorch介绍

安装 PyTorch 时，可以选择在 CPU 或 GPU 上运行，取决于你的硬件支持和需求。

CPU版本安装：

CPU 版本适用于没有 GPU 或不需要使用 GPU 进行深度学习的情况。安装过程相对简单，因为没有依赖 GPU 驱动和 CUDA 的要求。

GPU版本安装：

如果你的计算机上有 NVIDIA GPU，并且你希望加速深度学习计算，可以安装 GPU 版本的 PyTorch。GPU 版本依赖于 CUDA 和 cuDNN。我的电脑是M3 Mac Pro，虽然有GPU，但是不是NVIDIA GPU，如果要启用GPU的能力，需要从源代码编译PyTorch，并确保安装了必要的依赖项。可以看我另外一篇文章：星图模型：Mac电脑安装支持GPU加速的PyTorch版本

三、CPU版本PyTorch安装

为了更好的管理不同项目的Python项目，通常建议创建一个虚拟环境。可以隔离不同项目的依赖项，避免项目之间的冲突。

3.1 创建PyTorch的虚拟环境

执行命令：

conda create -n pytorch-env python=3.12.2

3.2 激活PyTorch的虚拟环境

执行如下命令：

conda activate pytorch-env

3.3 镜像源安装PyTorch

首先，你先查看自己的conda环境中有哪些镜像源，执行命令：

conda config --show channel

添加图片注释，不超过 140 字（可选）

然后，把你的pytorch镜像源加入进来：

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch

再执行如下命令进行安装：

conda install pytorch torchvision torchaudio cpuonly

3.4 验证安装完成

执行：

conda list

执行后，可以看见pytorch已经安装

然后执行如下两条命令：

命令1：启动Python环境

python

命令2: 导入PyTorch并检查版本

import torch

命令3：检查版本

print(torch.__version__)