Reddit - r/MachineLearning

PyTorch model running 170x slower on T4 vs A100. What could cause a bottleneck this extreme? [D]

Performance Profile

Seeing a ~170ร— slowdown running a point-tracking model on an NVIDIA T4 compared to an A100. On A100 the tracker takes ~0.5 seconds per half-video. On T4 the same call takes ~85 seconds. Video is 47 frames at 256ร—256, batch 1.

I expect a meaningful gap between these cards, but 170ร— feels too large to explain by generational hardware differences alone.

Setup

  • Precision: pure FP32
  • Architecture: builds local 4D correlation volumes (dense matching between frames) followed by transformer layers for temporal context

Already Ruled Out

  • GPU is at 99% utilization during the call (via nvidia-smi)
  • Model is actually on GPU (torch.cuda.is_available() = True, device prints "cuda")
  • Enabling torch.backends.cudnn.benchmark = True had no effect
  • Same slowdown on two independent T4 machines, so it's not a driver/setup issue

Question

Given the architecture (4D correlations + transformers) and pure FP32 execution, what would cause a T4 to be this much slower than A100? What should I look for or profile first?

Comments

No comments yet. Start the discussion.