Halloween DealsAmazon USHalloween week deals: useful picks before month-endAmazon US: current deals, useful picks and tech finds.Check DealsSlow PC?RecommendedPC slow today? Run a repair scan before it gets worseResolve common Windows issues and optimize system performance.Scan NowHalloween DealsAmazon USOctober picks are moving - compare before checkoutAmazon US: current deals, useful picks and tech finds.See Picks×
FREEDOM251

Freedom251 › Blog › AI infrastructure

How to choose storage and networking for data-intensive AI workloads

By Freedom251 Team · · 10 min read

Special offer. See more information about Outbyte and uninstall instructions. Please review EULA and Privacy policy.

Choose storage and networking from measured workload behavior—not from the label “AI.” Record dataset and checkpoint sizes, read/write patterns, metadata activity, concurrency, latency, throughput, recovery targets and growth. Then assign each data class to the least expensive tier that still meets its performance and reliability requirements, and benchmark the complete path from storage through the network to the compute nodes.

In practice, object storage usually holds durable datasets and model artifacts, a parallel file system serves active POSIX jobs with high bandwidth or metadata demand, and block storage or local NVMe SSD provides provisioned disks and compute-local scratch. Those roles overlap by service and configuration, so validate the design with representative jobs rather than assuming one technology is universally best.

As an Amazon Associate I earn from qualifying purchases.

Start with workload measurements, not storage labels

“AI training” can mean a read-mostly image corpus, millions of tiny text files, a streaming feature pipeline, or a job that writes hundreds of terabytes of checkpoints. Each pattern stresses storage and networking differently. NVIDIA summarizes the planning problem plainly: “There is no ‘one-size-fits-all’ storage solution for AI-driven apps.”

Special offer. See more information about Outbyte and uninstall instructions. Please review EULA and Privacy policy.

Build a workload profile before selecting a service or appliance. Capture the following for both training and inference:

#1 Best Overall
Sale
Samsung SSD 990 PRO 2TB, PCIe 4.0 M.2 2280, Up to 7,450 MB/s | NVMe Internal Solid State Hard Drive for High End Computing, Gaming, and Heavy Duty Workstations, (MZ-V9P2T0B/AM)
  • MEET THE NEXT GEN: Consider this a cheat code; Our Samsung 990 PRO Gen4 SSD helps you reach near max performance with lightning-fast speeds; Whether you’re a hardcore gamer or a tech guru, you’ll get power efficiency built for the final boss
  • REACH THE NEXT LEVEL: Gen4 steps up with faster transfer speeds and high-performance bandwidth; With a more than 55% improvement in random performance compared to 980 PRO, it’s here for heavy computing and faster loading
  • THE FASTEST SSD FROM THE WORLD'S FLASH MEMORY BRAND: The speed you need for any occasion; With read and write speeds up to 7450/6900 MB/s you’ll reach near max performance of PCIe 4.0 powering through for any use
  • PLAY WITHOUT LIMITS: Give yourself some space with storage capacities from 1TB to 4TB; Sync all your saves and reign supreme in gaming, video editing, data analysis and more
  • IT’S A POWER MOVE: Save the power for your performance; Get power efficiency all while experiencing up to 50% improved performance per watt over the 980 PRO; It makes every move more effective with less consumption
  • Capacity: current logical data, replicas, intermediate files, checkpoints and expected annual growth.
  • Object and file shape: number of objects or files, median and largest size, directory depth, and whether files are packed into larger shards.
  • Access pattern: sequential or random reads, read-mostly versus write-heavy behavior, rereads across epochs, and whether workers open the same files simultaneously.
  • Performance: sustained aggregate bandwidth, per-client bandwidth, IOPS, operation rate, latency and tail latency under peak concurrency.
  • Metadata: create, stat, rename, list and delete rates. Millions of small files can become a metadata bottleneck even when bulk bandwidth is high.
  • Concurrency: number of workers, accelerators per worker, jobs sharing the system, and synchronized starts or restores.
  • Checkpoint behavior: bytes per checkpoint, interval, consistency requirements, write pause that can be tolerated and restore deadline.
  • Recovery and governance: acceptable recovery point and recovery time, versioning, retention, encryption, region or zone constraints and portability requirements.

Use a production-shaped benchmark. A storage specification measured by one client with a large sequential file does not predict a synchronized launch by thousands of workers. Measure the application path, including serialization, caching, network hops, authentication and throttling.

Match each data class to a storage interface

The interface determines semantics as much as performance. Object APIs expose durable, highly scalable objects; POSIX file systems provide directories, file locking and familiar open/read/write behavior; block devices present disks that the host or a clustered file system must format and manage.

Storage type Best fit Strengths to measure Common limitations or risks
Object storage Large source datasets, model weights, durable artifacts, asynchronous or tiered checkpoints Scalable capacity and durability, lifecycle policies, broad access from many services Object semantics may require application changes; small random operations, listing and tail latency can be unsuitable for an active POSIX workload; request and transfer charges need to be modeled
Parallel file system Active training data, synchronous checkpoints, shared POSIX data and metadata-intensive pipelines Low-latency file access, high aggregate throughput and concurrent metadata operations when correctly sized Higher operational or service cost, finite namespace and client limits, and possible zone or region placement constraints
Block storage or shared block pools Provisioned disks, databases, scratch volumes and workloads needing tunable IOPS or throughput Predictable provisioned performance and familiar block-device semantics Hosts must manage formatting, mounting, sharing and replication; performance and cost often scale with the provisioned tier rather than actual utilization
Local NVMe SSD Compute-local scratch, preprocessing spill, caches and performance-sensitive temporary data Very low local latency and high bandwidth without a network hop Often tied to the host lifecycle, limited in capacity, and not a durable system of record; endurance, cooling and server support must be verified

Google’s AI Hypercomputer guidance follows this division: object storage for scalable datasets and artifacts, a POSIX parallel file system for low-latency and metadata-concurrent activity, and shared block pools for compute and scratch needs. Treat that as a role-based pattern, not a ranking.

Special offer. See more information about Outbyte and uninstall instructions. Please review EULA and Privacy policy.

Separate hot, warm and durable data

Keeping every byte on the fastest tier wastes budget and can make recovery harder. Define a lifecycle for each data class:

  1. Active tier: place files or objects needed by the current job on the interface that meets its latency, bandwidth and metadata targets. Cache or stage data when repeated reads justify it.
  2. Durable source tier: retain the authoritative dataset, approved model versions and completed checkpoints in durable object storage or another suitable repository with versioning and access controls.
  3. Lower-cost retention: move inactive objects to cool or archive tiers with explicit retention and retrieval expectations. Azure guidance recommends Blob Storage lifecycle policies for this purpose.
  4. Deletion and expiry: set rules for temporary shards, failed checkpoints and intermediate outputs. Confirm that legal retention or reproducibility requirements are not being removed.

Azure’s guidance recommends a file system for active, job-specific “hot” data, identifying Managed Lustre when transfer time and latency are priorities, Azure NetApp Files when enterprise features or hybrid integration matter, and local NVMe/SSD file systems when performance is the overriding concern. It recommends moving inactive data to Blob Storage and applying Cool or Archive lifecycle policies. The same logic applies outside Azure: separate the active working set from the durable source of truth.

Rank #2
Sale
Kingston NV3 1TB M.2 2280 NVMe SSD | PCIe 4.0 Gen 4x4 | Up to 6000 MB/s | SNV3S/1000G
  • Ideal for high speed, low power storage
  • Gen 4x4 NVMe PCle performance
  • Up to 6,000MB/s read, 4,000MB/s write
  • Includes Acronis cloning software
  • 5-year limited warranty

Design checkpointing before you scale the cluster

Checkpointing is simultaneously a storage, network and orchestration problem. A synchronous save can pause every worker while one large write completes. During recovery, many accelerators may request the same model and optimizer state at once, creating a burst that overwhelms an otherwise adequate link.

Estimate the bytes that must be written

Count model parameters, optimizer state, gradients or momentum that are persisted, precision conversions, framework metadata, random-number state and the effect of sharding or replication. The resulting checkpoint size—not just parameter count—determines write time and retained capacity. Measure a real checkpoint from the framework and parallelism configuration you will deploy.

Free tools Windows power users keep installed

One-click scans. No signup required.

Special offer. See more information about Outbyte and uninstall instructions. Please review EULA and Privacy policy.

Choose an interval from lost-work tolerance

Short intervals reduce recomputation after a failure but increase write traffic and possible pauses. Long intervals reduce overhead but increase the recovery point. Microsoft Learn gives “every 500 iterations” as an example interval; it is guidance for a particular scenario, not a universal rule. Set the interval from the value of lost training time, observed write duration and failure expectations.

Use asynchronous or hierarchical persistence when supported

An asynchronous design lets workers continue while a staging process persists the checkpoint. A hierarchical design can write to fast local or parallel storage first, then copy to durable object storage. During restore, a leader or small set of nodes can read the durable copy and distribute it across the accelerator cluster instead of having every worker pull the same large object independently. AWS describes this leader-distribution pattern as a way to reduce external-storage load.

Do not call a checkpoint complete until the required consistency and durability boundary has been reached. Record a manifest, generation or transaction marker so a restart cannot combine files from different checkpoints. Test interrupted writes, partial copies and a restore while other jobs are using the network.

Rank #3
Sandisk Optimus 5100 500GB NVMe SSD, PCIe 4.0, M.2 2280 | Up to 6,600MB/s Read Speeds, Up to 5,600MB/s Write Speeds, Internal Solid State Drive - SDSP51500GAN
  • SPEED UP PROJECTS. Launch creator applications fast with uncompromising PCIe 4.0 read speeds up to 7,100MB/s,[2] (1TB and 2TB[1] models) and write speeds up to 6,700MB/s[2] (1TB[1]-4TB[1] models).
  • CREATE AND STORE MORE. Make more room for your 4K videos and high-resolution images with capacities from 500GB[1] up to 4TB[1] on M.2 2280 built with our trusted 8th generation SANDISK BiCS QLC 3D CBA NAND.
  • IT GOES WHERE YOU GO. With an all-new power efficient design, your drive delivers high performance with low power, giving you more time to be productive while on the go.
  • UNCOMPROMISED RELIABILITY. With up to 1,200 TBW[3] (4TB[1] model) endurance rating, your drive is designed for creators.
  • KEEP YOUR DRIVE UPDATED. Monitor your SSD’s performance and check for updates with the downloadable SANDISK Dashboard application.[5]

Design two network paths, not one

Training traffic has at least two distinct patterns:

Special offer. See more information about Outbyte and uninstall instructions. Please review EULA and Privacy policy.
  • Storage-to-compute traffic: workers read datasets, write checkpoints and fetch model artifacts. It is usually bursty at startup, epoch boundaries and restore.
  • Worker-to-worker traffic: gradient synchronization, all-reduce, parameter exchange and other collective operations. It can be latency-sensitive and continuous during each training step.

A network that is fast for one path can still bottleneck the other. Map the topology from storage service to switches, hosts and accelerators. Measure aggregate and per-node bandwidth, tail latency, packet loss, congestion, oversubscription and behavior when all clients start together. Keep storage and collective traffic on isolated or quality-of-service-controlled paths where the platform permits it.

Account for placement and transfer boundaries

Place compute and active storage in the same zone or region when the service’s terms and resilience requirements allow it. Model cross-zone and cross-region transfer, replication traffic and restore traffic explicitly. A design that meets a single-node benchmark can fail when thousands of clients cross a constrained link simultaneously.

Size for the recovery event

Normal training throughput may not be the largest demand. A failed job can cause every worker to reload a large checkpoint while other jobs continue. Set a recovery target, calculate the required aggregate read rate, then test it with the intended number of clients. AWS publishes modeled examples including a 125 TB checkpoint and a defined 4,000-accelerator scenario; those are scenario calculations, not generic industry benchmarks.

Provider patterns and stated limits

The following examples illustrate how major cloud services expose the roles above. Product names, limits, pricing and regional availability change, so confirm the configuration and current terms before procurement.

Special offer. See more information about Outbyte and uninstall instructions. Please review EULA and Privacy policy.
Rank #4
Sale
Samsung SSD 990 PRO 1TB, PCIe 4.0 M.2 2280, Up to 7,450 MB/s | NVMe Internal Solid State Hard Drive for High End Computing, Gaming, and Heavy Duty Workstations (MZ-V9P1T0B/AM)
  • HUGE SPEED BOOST: Get random read/write speeds that are 40%/55% faster than 980 PRO; Experience up to 1400K/1550K IOPS, while sequential read/write speeds up to 7,450/6,900 MB/s reach near the max performance of PCIe 4.0*
  • BREAKTHROUGH POWER EFFICIENCY: Use less power and get more performance; Enjoy up to 50% improved performance per watt over 980 PRO, plus optimal power efficiency with max PCIe 4.0 performance**
  • SMART THERMAL CONTROL: Samsung's own nickel-coated controller delivers effective thermal control; With its slim size, 990 PRO is a perfect fit for desktops and laptops that meet the PCI-SIG D8 standard***
  • THE CHAMPION MAKER: Up to 65% improvement in random performance enables faster loads for an ultimate gaming experience on PS5 and DirectStorage PC games****
  • SAMSUNG MAGICIAN SOFTWARE: Get the most out of your SSD with Samsung Magician's advanced yet intuitive optimization tools; Monitor drive health, protect valuable data, and receive important updates for your 990 PRO
Provider guidance Active data pattern Durable or scalable pattern Important qualification
Microsoft Azure Managed Lustre for low latency and minimized transfer time; Azure NetApp Files where enterprise, hybrid or customization needs dominate; local NVMe/SSD when compute-local performance is the priority Blob Storage with lifecycle policies that move inactive data to Cool or Archive tiers The “every 500 iterations” checkpoint example is illustrative, not a universal interval
Google Cloud Managed Lustre for POSIX and metadata-concurrent activity; Hyperdisk pools for shared capacity, throughput and IOPS Cloud Storage for large datasets and model artifacts; Rapid Bucket or Rapid Cache where the documented configuration fits Documentation states Rapid Bucket can deliver up to 15 TB/s and sub-millisecond latency for open files in its described zonal context. That is a provider limit, not an independent cross-cloud benchmark
AWS FSx for Lustre linked to S3 for low-latency, high-throughput file access, or direct S3 access through a VPC gateway endpoint S3 as the durable source, with staged or hierarchical distribution for restores AWS treats training-data access and node-to-node exchange as separate networking concerns; configure and measure both

A practical selection procedure

  1. Inventory the pipeline. Write down dataset size, object/file count and size distribution, read/write ratio, preprocessing, cacheability, worker count, checkpoint contents and interval, cold-start requirement and retention period.
  2. Set measurable targets. Define minimum aggregate and per-client bandwidth, acceptable latency and tail latency, metadata operations per second, checkpoint pause, restore time, durability and recovery point.
  3. Pick candidate interfaces. Test object storage for scalable durable data; a parallel file system for active POSIX or metadata-heavy jobs; block or local NVMe for scratch and provisioned disks. Keep more than one candidate when access semantics are uncertain.
  4. Build the tier map. Identify what is hot during a job, what can be cached or staged, what must remain durable and what can expire. Add lifecycle and version rules before data accumulates.
  5. Prototype checkpoint and restore. Produce a real checkpoint, measure write pause and durable commit time, then restore it with the full intended client count. Include interrupted writes and a failed leader or worker.
  6. Benchmark the complete network path. Run synchronized starts, steady-state training, checkpoint writes, restores and worker collectives. Record oversubscription, congestion and tail behavior, not just peak throughput.
  7. Calculate cost per useful result. Include active and durable storage, requests, transfer, caching, provisioned IOPS or throughput, replication, idle capacity and the compute time lost to I/O. Compare cost per completed training run or inference request, not only price per stored terabyte.
  8. Document the expansion path. Note how capacity, throughput, metadata performance, clients and network links grow independently. A design that works for one job may need a different tier or topology when jobs share the system.
Independent reader supportYour contribution helps us test, update, and keep practical guides available for everyone.Support on Ko-Fi

NVMe SSD compatibility checklist

Local NVMe SSD is attractive for scratch and hot caches, but “NVMe” alone does not establish suitability for a GPU server. Before ordering, verify:

  • PCIe generation, lane count, protocol and the server’s supported form factor.
  • Usable capacity after over-provisioning, filesystem layout and reserved space.
  • Write endurance for preprocessing, shuffle and checkpoint staging; sustained performance rather than a short burst specification.
  • Cooling and thermal throttling in the intended chassis and accelerator configuration.
  • Whether the drive is ephemeral on host replacement and how data is rebuilt or repopulated.
  • Operating-system, firmware, boot and monitoring support from the server platform.

Use local NVMe as a cache or scratch tier unless you have an explicit replication and recovery design. Keep authoritative data and completed artifacts on durable storage.

Failure modes to test before production

Workers are idle while storage appears “fast”

Check whether the benchmark used one client, large sequential files or a warm cache. Re-run with the real file count, synchronized workers, cold-start behavior and metadata operations. Packing tiny files into larger shards may reduce namespace pressure, but confirm that the framework can read the resulting format.

Training slows at every checkpoint

Measure application pause separately from backend write time. Consider asynchronous or hierarchical persistence, a faster staging tier, fewer simultaneous writers or a checkpoint interval derived from lost-work tolerance. Confirm that the durable commit marker is still honored.

Special offer. See more information about Outbyte and uninstall instructions. Please review EULA and Privacy policy.

Restore collapses the network

Look for every worker fetching the same artifact independently. Use staged distribution, a leader fan-out pattern or a cache, and reserve bandwidth for collective traffic. Test the recovery event while unrelated jobs run.

Best Value
WD_Black SN7100 1TB NVMe SSD - Gen4 PCIe, M.2 2280, Up to 7,250 MB/s Read Speed, Up to 6,900 MB/s Write Speed, Next Gen TLC 3D NAND, for Laptops, Handheld Gaming Devices - WDS100T4X0E
  • This product has been replaced by our latest generation. Please search for the SANDISK Optimus GX 7100 NVMe SSD
  • HIGH-OCTANE GAMING. Experience speeds up to 7,250MB/s read and 6,900MB/s write (1-2TB models), with up to 35% faster performance than previous generation.
  • PURPOSE-BUILT. Designed for serious on-the-go gamers, with a PCIe Gen4 interface and SANDISK’s next generation TLC 3D NAND.
  • MORE TIME TO CLEAR THAT CHECKPOINT. Built with laptops and handheld gaming devices in mind, with up to 100% more power efficiency over the previous generation.
  • DO MORE WITH DASHBOARD. Ensure your drive is optimized for prime performance with the downloadable WD_BLACK Dashboard (Windows only).

Costs rise unexpectedly

Break the bill into stored bytes, requests, transfer, provisioned performance, replication, cache misses and idle active-tier capacity. Lifecycle rules should have an owner, a retention rationale and a tested retrieval path; otherwise data tends to remain on the expensive tier indefinitely.

Restarted jobs see inconsistent files

Use immutable checkpoint directories or object generations, write a manifest only after all parts are durable, and make readers select a complete generation. Exercise an interruption between data-part writes and manifest publication.

What a good architecture looks like

A robust design commonly has a durable object repository for source data and approved artifacts; an active parallel file system or cache for the current job’s POSIX working set; local NVMe or block storage for scratch and staging; and a network plan that separates storage traffic from accelerator collectives. It measures normal training, synchronized startup, checkpointing and full-cluster restore with the same client count expected in production.

What’s actually slowing this PC down?

Pick the symptom - the matching free tool is one click away.

Special offer. See more information about Outbyte and uninstall instructions. Please review EULA and Privacy policy.

The right answer can still be a single service when the workload is simple. The decision is justified when measured latency, bandwidth, metadata rate, recovery behavior, compatibility and cost meet the target with room for growth—not because a particular storage category is fashionable.

Product prices and availability are accurate as of the date/time indicated and are subject to change. Any price and availability information displayed on Amazon at the time of purchase will apply.

AI infrastructurecheckpointsData StorageHPCMachine Learningnetworking

Leave a Reply

Your email address will not be published. Required fields are marked *

Special offer. See more information about Outbyte and uninstall instructions. Please review EULA and Privacy policy.

Recommended PC Tool
Recommended PC Tool
Crashes, No Sound, or Screen Glitches?Free driver scan
PC Slower Than It Used to Be?Free scan - under a minute

Two free Windows tools

One Free Minute Could Fix That PC

Before you go - each of these free tools takes about a minute and tackles what quietly slows a Windows PC down.

Special offer. View Outbyte info, uninstall instructions, EULA, and Privacy Policy.