首页 > 技术文章 > PyTorch的安装及学习资料

inchbyinch 2019-12-30 17:57 原文

PyTorch安装

按照PyTorch官网提示,一行命令即可搞定,可以根据自己的硬件驱动(比如CUDA)选择之前的版本。

唯一值得注意的是,如果下载过慢,很可能是由于官方下载源的原因,可以更换国内镜像。(之前下载WSL模块时也遇到下载超慢的问题,换上国内镜像源之后速度6的飞起。)

PyTorch学习路径

PyTorch的好处是:

  • 提供了类似numpy的tensor结构,可以在GPU上运行;
  • 提供了自动求导功能;
  • 文档清晰。

学习资料推荐:

推荐阅读