PyTorch环境安装

前言

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

一、PyTorch是什么?

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

二、安装步骤

1.安装Anaconda

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

https://www.anaconda.com/

PyTorch环境安装_第1张图片
点击DOWNLOAD,下载之后进行安装,看个人情况修改安装选项,正常情况一切默认就好,之后我们就可以在开始界面找到Anaconda3这个文件夹
PyTorch环境安装_第2张图片
我们进入Anconda Pormpt(anaconda3),会出现一个命令行程序
PyTorch环境安装_第3张图片
首先创建一个虚拟环境用到命令(将中文改为自己需要的)

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 你创建的虚拟环境名称

PyTorch环境安装_第4张图片
之后前面括号里面的base就会变成你创建的虚拟环境名称,这就说明你创建成功了

2.更新显卡驱动(无nvidia显卡跳过)

进入nvidia官网:

https://www.nvidia.cn/

点击驱动程序

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

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

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

进入Anconda Pormpt查看CUDA版本,输入:

nvidia-smi

PyTorch环境安装_第5张图片
CUDA Version所写的是cuda 驱动版本,要记住这个版本号,例如我的是11.7,之后我们进入PyTorch官网:

https://pytorch.org/

点击install
PyTorch环境安装_第6张图片
下拉进入选择区域
PyTorch环境安装_第7张图片
第一栏: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()

PyTorch环境安装_第8张图片
如果命令行回复True则安装完成,cpu版本不会回复True,会回复Fault

参考

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

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

你可能感兴趣的:(MachineLearning,pytorch,深度学习,python,机器学习)