Deep learning

This section includes example notebooks using two of the most popular deep learning libraries, TensorFlow and PyTorch.

Because deep learning models are data- and computation-intensive, distributed training can be important. This section also includes information about and examples of distributed deep learning using Horovod and spark-tensorflow-distributor.

• Get started with TensorFlow Keras in Databricks
  • Requirements
  • Example notebook
• TensorFlow
  • TensorFlow versions included in Databricks Runtime ML
  • Install TensorFlow
  • TensorBoard
  • Use TensorFlow on a single node
• PyTorch
  • Install PyTorch
  • Use PyTorch on a single node
• Distributed training
  • Horovod
  • spark-tensorflow-distributor
• Deep Learning Pipelines
  • Migration guide to Databricks Runtime 7.0 ML and above