Model Inference using PyTorch
The notebook below follows our recommended inference workflow. This example demonstrates how to do model inference using PyTorch with a pre-trained ResNet-50 model and image files as input data.
The PyTorch Init Script notebook creates an init script named
pytorch-init.sh that installs required libraries for PyTorch. If you run on Databricks Runtime 5.1 ML (Beta) or above, you do not need to create the PyTorch init script and configure your cluster with the script.
Before running the Model inference with PyTorch notebook you must:
- If you are running Databricks Runtime 5.0 ML (Beta), do the following:
a. Run the PyTorch Init Script notebook.
b. Configure a cluster with the