PyTorch

PyTorch项目是一个Python包,提供GPU加速的张量计算和高水平的功能构建深度学习网络。许可的细节,请参阅PyTorch许可证医生在GitHub上。

监控和调试PyTorch模型,考虑使用TensorBoard。

PyTorch砖中包含运行时机器学习。如果您使用的是砖运行时,看到的安装PyTorch安装PyTorch指令。

请注意

这不是一个全面PyTorch指南。有关更多信息,请参见PyTorch网站。

单节点和分布式训练

测试和迁移单机工作流程,使用单节点集群。

深度学习分布式训练选项,请参阅分布式训练。

例如笔记本电脑

PyTorch笔记本

在新标签页打开笔记本

安装PyTorch

砖运行时为毫升

砖运行时机器学习包括PyTorch所以你可以创建集群和开始使用PyTorch。PyTorch安装在砖运行时版本的ML版本使用,看到发布说明。

砖运行时

砖建议您使用PyTorch列入砖运行时机器学习。但是,如果你必须使用砖运行时,PyTorch可以安装砖PyPI图书馆。下面的例子展示了如何安装PyTorch 1.5.0:

  • 在GPU集群、安装pytorch和torchvision通过指定以下:

    • 火炬= = 1.5.0

    • torchvision = = 0.6.0

  • 对CPU集群、安装pytorch和torchvision通过使用下面的轮子文件:

    https://download.pytorch.org/whl/cpu/torch-1.5.0%2Bcpu-cp37-cp37m-linux_x86_64.whl https://download.pytorch.org/whl/cpu/torchvision-0.6.0%2Bcpu-cp37-cp37m-linux_x86_64.whl