Mistral AI GPU OOM error during batch inference
Problem – Mistral AI GPU OOM during Batch Inference When running a data‑pipeline that performs text‑classification with the Mistral‑7B model on a GPU with 24 GiB VRAM (e.g., RTX 3090), the job crashes after processing a few 512‑token batches. The failure manifests as: torch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 3.12 GiB (GPU 0; 24.00 GiB … Read more