Azure Storage Connector for PyTorch Documentation

The Azure Storage Connector for PyTorch (azstoragetorch) is a library that provides seamless, performance-optimized integrations between Azure Storage and PyTorch. Use this library to easily access and store data in Azure Storage while using PyTorch. The library currently offers:

• File-like object for saving and loading PyTorch models (i.e., checkpointing) with Azure Blob Storage

• PyTorch datasets for loading data samples from Azure Blob Storage

Visit the Getting Started page for more information on how to start using the Azure Storage Connector for PyTorch.

User Guide

• User Guide
  • Getting Started
    • Prerequisites
    • Installation
    • Configuration
  • Saving and Loading PyTorch Models (Checkpointing)
    • Saving a Model
    • Loading a Model
  • PyTorch Datasets
    • Create Dataset from Azure Storage Container
    • Create Dataset from List of Blobs
    • Transforming Dataset Output
    • Using Dataset with PyTorch DataLoader
      • Iterable-style Datasets with Multiple Workers

API Reference

• API Reference
  • BlobIO
  • Datasets
  • Exceptions