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版本。
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/
如果有以下报错,表示该库文件可能不存在。
可以在conda官网上查询该库文件的最新版本
这个页面中有多种安装方式,conda或者Wheel,如果不支持联网,也可以进入到Wheel 的网址,将其手动下载下来后,使用离线安装方式。
2.3.2 离线安装
离线安装时,需要下载两个文件 torchvision 及 torch,其对应的版本应与安装指令中相同。因为我的电脑只有CPU,所以都下载的是CPU版本
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
引入 torch,打印出 torch 的版本号