PyTorch 简介
PyTorch 是一个开源的 Python 机器学习库,基于 Torch,用于自然语言处理等应用程序。
相关链接:
本文初次撰写于 2020-09-28,最后更新于 2020-09-29.
安装 PyTorch 之前
安装 Anaconda
Anaconda 是一个集成了大量用于科学分析(包括机器学习,深度学习等)的 package 的工具包,可以帮助我们管理相关的 package,是在单台机器上执行机器学习的最简单方法。
本文采用 Anaconda 作为 PyTorch 包的管理程序,您也可以选择 pip 等工具进行管理,不再赘述。
您可以点击此链接前往 Anaconda 的官方网站,下载并安装适用于个人用户学习使用的开源版程序。
安装成功后,您可以通过 Windows 10 的搜索功能或开始菜单轻松找到 Anaconda Powershell Prompt 或 Anaconda Prompt. 两者打开方式本质上没有区别,您可以任意选择。本文的后续演示将使用 Anaconda Powershell Prompt 进行操作。
假如您有 NVIDIA 系列显卡
假如没有,您可以跳过这一步,后续 PyTorch 的安装过程不会受到影响。
检验驱动程序
显卡可以大幅度提升机器学习时的处理速度。打开任务管理器 - 性能 - GPU, 假如您已经安装显卡驱动,那么会在此处(图中蓝框勾出)显示具体的显卡型号。假如您确实有 NVIDIA 显卡但在此处没有显示,可以前往 NVIDIA 官网下载对应的驱动程序并安装。
安装 CUDA
CUDA 是用于GPU编程的语言ÿ