PyTorch项目实战01——开发准备
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,可在官网下载 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…
如果有以下报错,表示该库文件可能不存在。
可以在conda官网上查询该库文件的最新版本
这个页面中有多种安装方式,conda或者Wheel,如果不支持联网,也可以进入到Wheel 的网址,将其手动下载下来后,使用离线安装方式。
2.3.2 离线安装
离线安装时,需要下载两个文件 torchvision 及 torch,其对应的版本应与安装指令中相同。因为我的电脑只有CPU,所以都下载的是CPU版本
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库并能正确打印出版本信息后,表示安装成功。
转载自:https://juejin.cn/post/7224081662468210749