Prerequisites#
The following hardware and software is required to perform inference and post-training with Cosmos-Predict1.
Hardware#
We recommend using NVIDIA H100-80GB or A100-80GB GPUs for inference and post-training. Refer to the Model Matrix page for the required number of GPUs for inference and post-training.
Software#
Ubuntu: 24.04, 22.04, or 20.04
Python: 3.10.x with conda installed
NVIDIA Container Toolkit: 1.16.2 or later