Can you run cross-encoder/stsb-roberta-large on A100 80GB?

cross-encoder/stsb-roberta-large needs ~12.99 GB, which is within the 80 GB on A100 80GB by memory capacity (~67.01 GB headroom). Computed from stated size and quant, not a runtime guarantee.

cross-encoder/stsb-roberta-large needs ~12.99 GB, which is within the 80 GB on A100 80GB by memory capacity (~67.01 GB headroom). Computed from stated size and quant, not a runtime guarantee.

Memory-capacity estimate across common GPUs

GPUVerdictVRAMNeeds
8GB laptop (no dedicated GPU)Does not fit8 GB~12.99 GB
16GB laptop (no dedicated GPU)Fits (by memory)16 GB~12.99 GB
NVIDIA T4 16GB (free Colab)Fits (by memory)16 GB~12.99 GB
NVIDIA L4 24GBFits (by memory)24 GB~12.99 GB
RTX 3060 12GBDoes not fit12 GB~12.99 GB
RTX 4080 16GBFits (by memory)16 GB~12.99 GB
RTX 3090 24GBFits (by memory)24 GB~12.99 GB
RTX 4090 24GBFits (by memory)24 GB~12.99 GB
A100 40GBFits (by memory)40 GB~12.99 GB
A100 80GBFits (by memory)80 GB~12.99 GB
H100 80GBFits (by memory)80 GB~12.99 GB
Apple M-series 16GB (unified)Fits (by memory)16 GB~12.99 GB
Apple M-series 32GB (unified)Fits (by memory)32 GB~12.99 GB
Apple M-series 64GB (unified)Fits (by memory)64 GB~12.99 GB
Apple M-series 128GB (unified)Fits (by memory)128 GB~12.99 GB
CPU / 32GB system RAMFits (by memory)32 GB~12.99 GB
CPU / 64GB system RAMFits (by memory)64 GB~12.99 GB

How this was computed

  • Weights + KV cache at the stated context + a runtime/overhead allowance; memory capacity only, not a benchmark.

Next steps

Methodology: weights + KV-cache at 8192 tokens + a runtime overhead allowance, from the recommended download size, parameter count, and quant. Computed from stated size and quant, not a runtime guarantee. Not a benchmark; runtimes differ.