Is this page helpful?

Support Matrix#

Supported Hardware#

Minimum System Hardware Requirements#

CPU Cores

RAM (GB)

NVMe SSD Storage (GB)

GPU Count

4

8

10

1 GPU from the list of Supported NVIDIA GPUs

Note

The NVMe SSD Storage requirement of 10GB 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. We recommend using one of the following tested configurations:

GPU

GPU Memory (GB)

Microarchitecture

Precision

GPU Count

B200

192

Blackwell

FP32/FP16

1

GB200

192

Blackwell

FP32/FP16

1

RTX PRO 6000 Blackwell Workstation Edition

96

Blackwell

FP32/FP16

1

GB10

128

Blackwell

FP32/FP16

1

GH200

144

Hopper

FP32/FP16

1

H200

141

Hopper

FP32/FP16

1

H100

80

Hopper

FP32/FP16

1

A100

40-80

Ampere

FP32/FP16

1

L40S

48

Ada Lovelace

FP32/FP16

1

Note

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

Testing Locally Available Hardware#

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      |
|=============================================================================|
+-----------------------------------------------------------------------------+