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