在Google TPU上启用PyTorch-Python开发

时间:2024-06-14 08:44:26
【文件属性】:

文件名称:在Google TPU上启用PyTorch-Python开发

文件大小:752KB

文件格式:ZIP

更新时间:2024-06-14 08:44:26

Python Deep Learning

PyTorch / XLA是一个Python软件包,它使用XLA深度学习编译器来连接PyTorch深度学习框架和Cloud TPU。 您可以立即在带有Google Colab的单个Cloud TPU上免费试用,并在生产环境中使用,并在带有Google Cloud的Cloud TPU Pod上使用。 PyTorch / XLA当前的CI状态:PyTorch / XLA是一个Python软件包,它使用XLA深度学习编译器连接PyTorch深度学习框架和Cloud TPU。 您可以立即在带有Google Colab的单个Cloud TPU上免费试用,并在生产环境中使用,并在带有Google Cloud的Cloud TPU Pod上使用。 看看我们的Colab笔记本之一,以快速尝试在Cloud TPU上运行的不同PyTorch网络,并学习如何将Cloud TPU用作PyTorch设备:PyTorch入门o


网友评论