PyTorch 01:PyTorch 简介与安装

PyTorch 学习笔记

Posted by YEY on December 5, 2020

Lecture 01 PyTorch 简介与安装

1. PyTorch 简介

1.1 PyTorch 简介

2017 年 1 月, FAIR (Facebook AI Research) 发布 PyTorch,它是在 Torch 基础上用 Python 语言重新打造的一款深度学习框架。Torch 是采用 Lua 语言为接口的机器学习框架,但因 Lua 语言较为小众,导致 Torch 知名度不高。

1.2 PyTorch 发展

  • 2017 年 1 月正式发布 PyTorch。
  • 2018 年 4 月更新 0.4.0 版,支持 Windows 系统,caffe2 正式并入 PyTorch。
  • 2018 年 11 月更新 1.0 稳定版,已成为 GitHub 上增长第二快的开源项目。
  • 2019 年 5 月更新 1.1.0 版,支持 TensorBoard,增强可视化功能。
  • 2019 年 8 月更新 1.2.0 版,更新 torchvision,torchaudio 和 torchtext,增加更多功能。

2014 年 10 月至 2018 年 2 月 arXiv 论文中深度学习框架提及次数统计:

PyTorch 的增长速度与 TensorFlow 一致。

2019 年 3 月各深度学习框架在 GitHub 上的 Stars、Forks、Watchers 和 Contributors 数量对比:

1.3 PyTorch 优点

  • 上手快:掌握 Numpy 和基本深度学习概念即可上手。
  • 代码简洁灵活:用 nn.module 封装使网络搭建更方便;基于动态图机制,更灵活。
  • Debug 方便:调试 PyTorch 就像调试 Python 代码一样简单。
  • 文档规范https://pytorch.org/docs/ 可查各版本文档。
  • 资源多:arXiv 中的新算法大多有 PyTorch 实现。
  • 开发者多:GitHub 上贡献者 (Contributors) 已超过 1100+。
  • 背靠大树:FaceBook 维护开发。
  • ……

1.4 适合人群

  • 深度学习初学者:模型算法实现容易,加深深度学习概念认识。
  • 机器学习爱好者:数十行代码便可实现人脸识别、目标检测、图像生成等有趣实验。
  • 算法研究员:最新 arXiv 论文算法快速复现。

2. 软件安装

2.1 软件安装

  • Python 包管理器:Anaconda。
  • Python 集成开发环境:PyCharm。
  • 深度学习框架:PyTorch。

2.2 解释器与工具包

  • 工具包
    • 工具包 又称为 依赖包模块
    • Python 之所以强大是因为拥有大量工具包。
    • 内置包ossysglobremath 等。
    • 第三方包pytorchtensorflownumpy 等。

2.3 虚拟环境

2.4 Anaconda 安装

Anaconda 是为方便使用 Python 而建立的一个软件包,其包含常用的 250 多个 工具包,多版本 Python 解释器 和强大的 虚拟环境管理 工具,所以 Anaconda 也被称为 Python 全家桶。

Anaconda 可以使安装、运行和升级环境变得更简单,因此推荐安装使用。

安装步骤

  1. 官网下载安装包:https://www.anaconda.com/distribution/#download-section
  2. 运行 Anaconda3-2019.07-Windows-x86_64.exe
  3. 选择路径,勾选 Add Anaconda to the system PATH environment variable,等待安装完成
  4. 验证安装成功,打开终端,输入 conda,回车
  5. 添加中科大镜像

设置 conda 国内镜像源(加速)

打开终端,复制以下命令,回车:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
# 查看 conda 配置
conda config --show

# 添加 conda 中科大镜像源
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 清华镜像源
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

# 设置搜索时显示通道地址
conda config --set show_channel_urls yes

# 重置 conda channels
conda config --remove-key channels

创建 conda 虚拟环境

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
# 查看 conda 虚拟环境
conda env list

# 创建 conda 虚拟环境, 由于网络原因可能会创建失败,可以通过多次尝试或者切换镜像源解决
# 环境名为 "pytorch_1.5_cpu", 解释器为 Python 3.6
conda create -n pytorch_1.5_cpu python=3.6

# 启动 conda 虚拟环境
conda activate pytorch_1.5_cpu

# 退出 conda 虚拟环境, 如果是在一个虚拟环境下启动另一个虚拟环境,需要执行两次退出回到 base 环境
conda deactivate

# 删除 conda 虚拟环境
conda env remove -n pytorch_1.5_cpu

2.5 Pycharm 安装

Pycharm 是由捷克公司 JetBrains 开发的一款强大的 Python IDE,拥有 调试语法高亮、Project 管理、代码跳转智能提示、版本控制等功能。

安装步骤

  1. 官网下载安装包:https://www.jetbrains.com/pycharm/
  2. 运行 pycharm-professional-2019.2.exe
  3. 选择路径,勾选 Add launchers dir to the PATH,等待安装完成

2.6 PyTorch 安装

安装步骤

  1. 检查是否有合适 GPU,若有,需安装 CUDACuDNN
  2. CUDACuDNN 安装(非必须)
  3. 下载 whl 文件,登陆 https://download.pytorch.org/whl/torch_stable.html

    命名解释:

    下载 pytorch 与 torchvision 的 whl 文件,进入相应虚拟环境,通过 pip 安装

  4. 在 pycharm 中创建 hello pytorch 项目,运行脚本, 查看 pytorch 版本

    1
    2
    3
    
    import torch
    print("hello pytorch {}".format(torch.__version__))
    print("cuda is avaiable: {}".format(torch.cuda.is_available()))
    

3. 总结

本节课介绍了 PyTorch 及其作为深度学习框架的优势,并且对安装 Pycharm、Anaconda、Cuda、cudnn 和 PyTorch 进行了演示,环境配置好后会进行 demo 演示,测试 PyTorch 可以正常使用。

下节内容:张量简介及创建

知识共享许可协议本作品采用知识共享署名-非商业性使用-相同方式共享 4.0 国际许可协议进行许可。 欢迎转载,并请注明来自:YEY 的博客 同时保持文章内容的完整和以上声明信息!