PyTorch安装教程：从环境配置到测试运行

PyTorch安装教程：从环境配置到测试运行

PyTorch是深度学习领域中非常流行的框架之一，为了帮助大家顺利安装和使用PyTorch，本文将详细介绍从环境配置到安装测试的完整流程。

下载PyTorch

访问官方网站：https://pytorch.org/

下拉页面，找到如下图所示的样式。

查看电脑GPU版本

有以下两种方法可以查看GPU版本：

方法1：使用命令行

1. 按住Win + R键，输入cmd打开命令提示符。
2. 在命令提示符中输入nvidia-smi。
3. 查看输出结果中的GPU版本号。例如，我的GPU版本号是12.2。

方法2：通过NVIDIA控制面板

1. 在电脑上搜索nvidia。
2. 打开NVIDIA控制面板，点击帮助 -> 系统信息。
3. 在弹出的界面中点击组件，查看GPU版本信息。例如，我的GPU版本是12.2.146。

回到PyTorch官网（https://pytorch.org/），根据查询到的GPU版本选择合适的配置。蓝色框线标注的部分适合我的GPU版本。

创建虚拟环境

1. 在开始菜单中找到Anaconda3(64-bit)。
2. 点击红色框线标注的图标，打开Anaconda Prompt。
3. 输入以下命令创建新的虚拟环境：

conda create -n 自己环境的名字 python=x.x

例如，如果要创建一个基于Python 3.9的PyTorch环境，可以命名为pytorch_py39。

激活虚拟环境

在Anaconda Prompt中输入以下命令激活刚刚创建的环境：

activate 自己环境的名字

安装PyTorch库

1. 在PyTorch官网根据你的GPU版本选择合适的配置。如果没有GPU，可以选择CPU版本。
2. 系统会自动生成相应的安装代码，通常是一个pip安装命令。
3. 将生成的代码复制到已经激活的虚拟环境中。
4. 系统会提示是否继续安装，输入y确认，等待安装完成，直到显示done。

测试安装是否成功

1. 激活你的虚拟环境：

activate 自己环境的名字

2. 进入Python环境：

python

3. 输入以下命令测试PyTorch是否正确安装：

import torch

如果没有任何错误提示，说明PyTorch已经成功安装。