专栏名称: 量子位
վ'ᴗ' ի 追踪AI行业和技术动态,这里更快一步!关注我们,回复“今天”,更多大新闻等你来发现
今天看啥  ›  专栏  ›  量子位

一行代码安装,TPU也能运行PyTorch,修改少量代码即可快速移植

量子位  · 公众号  ·  · 2020-02-28 14:07
晓查 发自 凹非寺 量子位 报道 | 公众号 QbitAI对于PyTorch开发者来说,Google Colab上的TPU资源不能用,恐怕是最遗憾的事情了。Google的云计算资源眼看着不能用,转TensorFlow又是不可能的。过去一直有PyTorch用户试图在Colab上薅羊毛,但是都没有太成功的。现在福利来了,一个叫做Pytorch Lightning的项目,可以让你几乎修改代码的情况下用上TPU。Pytorch Lightning已经上传到PyPI,因此只需一行代码就能安装这个软件。pip install pytorch-lightning该项目的开发者William Falcon说,PyTorch Lightning是他在纽约大学和FAIR做博士生时研发,专门为从事AI研究的专业研究人员和博士生创建的。几乎无需修改代码首先让我们来看一个MNIST图像分类网络的搭建,PyTorch的原始代码和修改后的PyTorch Lightning代码几 ………………………………

原文地址:访问原文地址
快照地址: 访问文章快照