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
  • Single node and distributed training
  • Example notebook
• TensorFlow
  • Single node and distributed training
  • Example notebook
  • TensorBoard
  • Install or downgrade TensorFlow
• PyTorch
  • Single node and distributed training
  • Example notebook
  • Install PyTorch
• Distributed training
  • Horovod
  • spark-tensorflow-distributor
• Deep Learning Pipelines
  • Migration guide to Databricks Runtime 7.0 ML and above