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——开发准备_python

2.2 安装Python

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

首先下载Anaconda,可在官网下载 https://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官网 https://pytorch.org/,查看适合自己安装的版本。

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

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

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

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

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

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


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

2.3.2 离线安装

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

PyTorch项目实战01——开发准备_虚拟环境_05

2.3.3 环境验证

验证并打印出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库并能正确打印出版本信息后,表示安装成功。

2.4 在Jupyter中安装Torch

安装ipykernel,进入到 conda 中pytorch虚拟环境中,运行以下命令安装 ipykernel

(pytorch12) D:\>pip install ipykernel

安装pytorch

(pytorch12) D:\>python -m ipykernel install --user --name pytorch12 --display-name "pytorch"

打开jupyter并使用 pytorch 创建新的文件

jupyter notebook

PyTorch项目实战01——开发准备_虚拟环境_06

引入 torch,打印出 torch 的版本号

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