-
PyTorch是一个基于Torch的Python开源机器学习库,用于计算机视觉等应用程序。 它主要由Facebook的人工智能研究小组开发。
-
PyTorch是一个Python包,提供两个高级功能: 具有强大的GPU加速的张量计算(如NumPy) 包含自动求导系统的的深度神经网络。
PyTorch算是相当简洁优雅且高效快速的框架
设计追求最少的封装,尽量避免重复造轮子
面向对象设计,设计最符合人们的思维,它让用户尽可能地专注于实现自己的想法
与google的Tensorflow类似,FAIR的支持足以确保PyTorch获得持续的开发更新
不错的的文档(相比FB的其他项目,PyTorch的文档简直算是完善了,参考Thrift),PyTorch作者亲自维护的论坛 供用户交流和求教问题
-
PyTorch的安装十分简单,根据PyTorch官网,对系统选择和安装方式等灵活选择即可。 国内安装anaconda建议使用清华镜像。
-
pytorch的安装经过了几次变化,请大家以官网的安装命令为准。另外需要说明的就是在1.2版本以后,pytorch只支持cuda 9.2以上了,所以需要对cuda进行升级,目前测试大部分显卡都可以用,包括笔记本的MX250也是可以顺利升级到cuda 10.1。
-
# 安装CPU版本的torch,手上的电脑为notebook 1050,没找到官网对应的cuda加速包和cudatoolkit pip install torch torchvision cpuonly
-
# 验证 import torch print(torch.__version__)
-
新建的环境是没有安装 ipykernel 的, 所以无法注册到Jupyter Notebook中,先要准备下环境(新建一个虚拟环境来管理自己的pytorch环境_羞儿的博客-CSDN博客)
-
#安装ipykernel conda install ipykernel conda install jupyter
-
基于Python的科学计算包,服务于以下两种场景:
-
作为NumPy的替代品,可以使用GPU的强大计算能力
-
提供最大的灵活性和高速的深度学习研究平台
-