文件名称:在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