PyTorch环境安装

前言

本教程适用于nvidia显卡或无nvidia显卡的电脑

一、PyTorch是什么？

PyTorch是一个开源的Python机器学习库，基于Torch，用于自然语言处理等应用程序。

二、安装步骤

1.安装Anaconda

我们需要安装Anaconda来对各种包来进行管理，进入Anaconda官网

https://www.anaconda.com/

点击DOWNLOAD，下载之后进行安装，看个人情况修改安装选项，正常情况一切默认就好，之后我们就可以在开始界面找到Anaconda3这个文件夹

我们进入Anconda Pormpt(anaconda3),会出现一个命令行程序

首先创建一个虚拟环境用到命令（将中文改为自己需要的）

conda create --name  你创建的虚拟环境名称  python=你需要安装python版本 

创建的同时需要安装一些依赖，需要确定按y回车确定就可以继续安装了
如果无法创建我们需要将Anaconda源换为中国的源，输入以下命令：
换清华源：

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes

换中科大源：

conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/cloud/bioconda/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/cloud/menpo/
conda config --set show_channel_urls yes

然后再次运行此命令：

conda create --name  你创建的虚拟环境名称  python=你需要安装python版本 

创建完成后，输入命令：

activate 你创建的虚拟环境名称

之后前面括号里面的base就会变成你创建的虚拟环境名称，这就说明你创建成功了

2.更新显卡驱动（无nvidia显卡跳过）

进入nvidia官网：

https://www.nvidia.cn/

点击驱动程序

选择你电脑的GPU型号和操作系统版本（如果是笔记本就选择NOTEBOOK系列）

如果不知道自己GPU是什么型号进入任务管理器—>性能—>GPU 查看
之后点击搜索，下载驱动安装程序，下载完成之后打开安装，将显卡驱动程序更新到最新

3.确定自己要安装的Pytorch版本（无nvidia显卡直接选择CPU版本）

进入Anconda Pormpt查看CUDA版本，输入：

nvidia-smi

CUDA Version所写的是cuda 驱动版本，要记住这个版本号，例如我的是11.7，之后我们进入PyTorch官网：

https://pytorch.org/

点击install

下拉进入选择区域

第一栏：PyTorch Build是选择下载类型分别是Stable(稳定版)，Preview(预览版)，LTS（长期版），我们一般选择Stable(稳定版)
第二栏：Your OS选择你电脑操作系统
第三栏：Language 选择你编程的编程语言，一般我们用python，看个人需求
第四栏：Compute Platform选择你的平台，现在选择的版本要<=我们刚才记得版本号，一般选择最新的，例如我的cuda驱动版本号是11.7，而最新的平台是11.6，所以我选择11.6版本，无nvidia显卡直接选择CPU版本
最后一栏：Run this Command就要复制下来这一条命令，这条命令是安装指令

4.安装对应的Pytorch版本

进入Anconda Pormpt，进入之前的创建的虚拟空间：

activate 你创建的虚拟环境名称

将复制的安装指令复制进来，回车安装，同样遇到需要确认的，按y回车确认安装，安装需要一段时间

5.验证Pytorch安装

进入Anconda Pormpt，进入之前的创建的虚拟空间：

activate 你创建的虚拟环境名称

输入python
进入python命令行
分别输入：

import torch
torch.cuda.is_available()

如果命令行回复True则安装完成，cpu版本不会回复True，会回复Fault

参考

B站UP主 我是土堆 链接：
https://www.bilibili.com/video/BV1S5411X7FY?vd_source=b69bbe200f14d370d3bd3ee02a48cd69

最后，如果有错误还请各位指出