Pytorch:简介和安装


PyTorch 简介

PyTorch 是一个开源Python 机器学习库,提供两个高级功能:1、具有强大的GPU加速张量计算;2、包含自动求导系统深度神经网络

PyTorch 的优点:简洁且高效快速、设计符合人类的思维、入门简单、FAIR的支持保证可以持续更新、作者亲自维护交流论坛。已兼容 Windows(CUDA,CPU)、MacOS(CPU)、Linux(CUDA,ROCm,CPU)。

:框架只是给我们提供了轮子(工具),让我们造汽车(产品)更加方便,最重要的还是我们个人的科学素养的提升(设计)。

PyTorch 的安装

登录 PyTorch官网 → 根据自己情况选择命令并复制(推荐使用 Anaconda 配套)→ 运行命令

检查是否安装成功:输入python → 输入import torch,没有报错则安装成功

注:Anaconda、Pycharm 和 Jupyter 是常用的组合,相关安装教程可在他处寻找,本文不再赘述。

显卡检查

使用任务管理器查看自己是否有 NVIDIA 的独立显卡及其型号。

NVIDIA GeForce GT 620 属于入门级独立显卡、低端显卡,较差。

BUG总结

输入 Pytorch 的安装命令后报错,如下图所示:

按照提示,手动删除对应的文件和文件夹;再输入安装pytorch的命令即可。

PyTorch 相关资源

  • Awesome-pytorch-list:目前已获12K Star,包含了NLP,CV,常见库,论文实现以及Pytorch的其他项目。
  • PyTorch官方文档:官方发布的文档,十分丰富。
  • Pytorch-handbook:GitHub上已经收获14.8K,pytorch手中书。
  • PyTorch官方社区:在这里你可以和开发pytorch的人们进行交流
  • 深入浅出 PyTorch:Datawhale 官方教程