Skip to navigation

Support Matrix

View as Markdown

Supported Hardware

Minimum System Hardware Requirements

CPU CoresRAM (GB)NVMe SSD Storage (GB)GPU Count
48101 GPU from the list of Supported NVIDIA GPUs

The NVMe SSD Storage requirement of 10 GB accounts for the ProteinMPNN NIM image size plus additional space for Docker layer caching and temporary files during runtime.

Supported NVIDIA GPUs

The ProteinMPNN NIM requires NVIDIA GPUs with at least 3 GB of GPU Memory. The ProteinMPNN NIM is configured to run on a single GPU and should run on any NVIDIA GPU that meets this minimum memory requirement and has compute capability >7.0. The following configurations have been specifically tested for optimal performance and stability. Use one of the following tested configurations:

GPUGPU Memory (GB)MicroarchitecturePrecisionGPU Count
A10040-80AmpereFP32/FP161
B200192BlackwellFP32/FP161
B300288BlackwellFP32/FP161
GB10128BlackwellFP32/FP161
GB200192BlackwellFP32/FP161
GB300288BlackwellFP32/FP161
GH200144HopperFP32/FP161
H10080HopperFP32/FP161
H200141HopperFP32/FP161
L40S48Ada LovelaceFP32/FP161
RTX 6000 Ada Generation48Ada LovelaceFP32/FP161
RTX PRO 6000 Blackwell Workstation Edition96BlackwellFP32/FP161

While the minimum GPU memory requirement is 3 GB, the above configurations have been tested for optimal performance and stability. Other NVIDIA GPUs with compute capability >7.0 might work but have not been officially tested.

Set Up the Software Environment

docker run --rm --runtime=nvidia --gpus 1 ubuntu nvidia-smi

Example output:

+-----------------------------------------------------------------------------+
| NVIDIA-SMI 580.126.09 Driver Version: 580.126.09 CUDA Version: 13.0 |
|-------------------------------+----------------------+----------------------+
| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
| | | MIG M. |
|===============================+======================+======================|
| 0 NVIDIA GeForce ... Off | 00000000:01:00.0 Off | N/A |
| 41% 30C P8 1W / 260W | 2244MiB / 11264MiB | 0% Default |
| | | N/A |
+-------------------------------+----------------------+----------------------+
+-----------------------------------------------------------------------------+
| Processes: |
| GPU GI CI PID Type Process name GPU Memory |
| ID ID Usage |
|=============================================================================|
+-----------------------------------------------------------------------------+