PyTorch

PyTorch項目是一個Python包，提供GPU加速張量計算和用於構建深度學習網絡的高級功能。有關許可的詳細信息，請參閱PyTorchGitHub上的許可文件．

要監視和調試您的PyTorch模型，請考慮使用TensorBoard．

PyTorch包含在機器學習的Databricks運行時中。如果您正在使用Databricks Runtime，請參見安裝PyTorch參閱安裝PyTorch的說明。

請注意

這不是PyTorch的全麵指南。有關更多信息，請參見PyTorch網站．

單節點和分布式訓練

要測試和遷移單機工作流，請使用單節點集群．

有關深度學習的分布式培訓選項，請參見分布式訓練．

例如筆記本電腦

PyTorch筆記本

在新選項卡中打開筆記本

安裝PyTorch

ML的Databricks運行時

用於機器學習的Databricks運行時包含PyTorch，所以你可以創建集群並開始使用PyTorch。有關您正在使用的Databricks Runtime ML版本中安裝的PyTorch的版本，請參閱發布說明．

磚運行時

Databricks建議您使用自帶的PyTorch用於機器學習的Databricks運行時．然而，如果你必須使用磚運行時， PyTorch可以安裝為Databricks 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