硬件和软件先决条件
在开始使用 BioNeMo 框架之前,请确保满足下面列出的硬件和软件先决条件。
硬件
BioNeMo 框架与可以访问 NVIDIA GPU 的环境兼容。bfloat16
精度需要 Ampere 一代或更高版本的 GPU (计算能力 ≥8.0)。您可能可以在没有 bfloat16
的 GPU 上运行 BioNeMo,但开发团队不支持这种情况。
GPU 支持矩阵
以下数据中心和桌面 GPU 的计算能力 ≥8.0,并且是 BioNeMo 的受支持硬件
GPU | 计算能力 | 支持 |
---|---|---|
H100 | 9.0 | 完整 |
L4 | 8.9 | 完整 |
L40 | 8.9 | 完整 |
A100 | 8.0 | 完整 |
A40 | 8.6 | 完整 |
A30 | 8.0 | 完整 |
A10 | 8.6 | 完整 |
A16 | 8.6 | 完整 |
A2 | 8.6 | 完整 |
RTX 6000 | 8.9 | 完整 |
RTX A6000 | 8.6 | 完整 |
RTX A5000 | 8.6 | 完整 |
RTX A4000 | 8.6 | 完整 |
软件
BioNeMo 框架在 x86 Linux 系统上受支持。
请确保在您期望的执行环境中安装了以下内容
- 适当的 GPU 驱动程序(最低版本:560;较低版本可能兼容,但不能保证)
- Docker(具有 GPU 支持,Docker Engine 19.03 或更高版本)
- NVIDIA Container Toolkit,以允许 Docker 访问 GPU