likes
comments
collection
share

PyTorch项目实战01——开发准备

作者站长头像
站长
· 阅读数 3

1 PyTorch概述

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

2017年1月,由Facebook人工智能研究院(FAIR)基于Torch推出了PyTorch。它是一个基于Python的可续计算包,提供两个高级功能:1、具有强大的GPU加速的张量计算(如NumPy)。2、包含自动求导系统的深度神经网络。

2 PyTorch环境安装

2.1 GPU or CPU

安装Pytorch前,需要先检查当前环境是否存在GPU,如果没有,就需要安装CPU版本;如果有GPU,则安装GPU版本。

打开设备管理器,查看“显示适配器”,

  • 如果存在 NVIDIA 字样的内容,即表示采用了独立显卡,支持GPU,可以安装GPU版本;
  • 如果和下图一样,仅有一行内容,即表示采用的是集成显卡,不支持GPU,需要安装CPU版本。

PyTorch项目实战01——开发准备

2.2 安装Python

安装 PyTorch 时,需要有python环境,因此需要先安装python,安装python时,优先建议采用 Anaconda 方式安装。

首先下载Anaconda,可在官网下载 anaconda.org/

安装完成后,可以为 PyTorch 运行设置独立的python运行环境:

conda create -n pytorch12 pythnotallow=3.8

表示创建运行PyTorch 的目录为 pytorch12,且为其指定 python版本为3.8。

创建好的目录可以使用如下命令查看

conda env list

显示当前存在的python虚拟环境列表

D:\log>conda env list

conda environments:

base                     D:\ProgramData\miniconda3

my_first_env             D:\ProgramData\miniconda3\envs\my_first_env

python_lib               D:\ProgramData\miniconda3\envs\python_lib

pytorch12                D:\ProgramData\miniconda3\envs\pytorch12

激活运行PyTorch的python虚拟环境

activate pytorch12

运行在选定的python虚拟环境中

(pytorch12) D:\log>

2.3 安装PyTorch

2.3.1 在线安装

接着进入到PyTorch官网 pytorch.org/, 查看适合自己安装的版本。

建议安装历史版本,因为如果有坑,可能都被前人趟过,容易找到解决方案,网址为 pytorch.org/get-started…

PyTorch项目实战01——开发准备

如果有以下报错,表示该库文件可能不存在。

PyTorch项目实战01——开发准备

可以在conda官网上查询该库文件的最新版本

PyTorch项目实战01——开发准备

这个页面中有多种安装方式,conda或者Wheel,如果不支持联网,也可以进入到Wheel 的网址,将其手动下载下来后,使用离线安装方式。

2.3.2 离线安装

离线安装时,需要下载两个文件 torchvision 及 torch,其对应的版本应与安装指令中相同。因为我的电脑只有CPU,所以都下载的是CPU版本

PyTorch项目实战01——开发准备

2.4 环境验证

验证并打印出PyTorch版本

(pytorch12) D:\log>python

Python 3.8.16 (default, Mar  2 2023, 03:18:16) [MSC v.1916 64 bit (AMD64)] :: Anaconda, Inc. on win32

Type "help", "copyright", "credits" or "license" for more information.

>import torch

>torch.version

'1.13.0+cpu'

当引用torch库并能正确打印出版本信息后,表示安装成功。

PyTorch项目实战01——开发准备

PyTorch项目实战01——开发准备