# create hopper-blackwell env

```
conda env create -f ./cosmos-predict2-hb.yaml 
# make sure torch in requirements-conda.txt is torch>=2.6.0
pip install -r requirements-conda.txt
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128
pip install flash-attn==2.6.3 --no-build-isolation
pip install -v --disable-pip-version-check --no-cache-dir --no-build-isolation --config-settings "--build-option=--cpp_ext --cuda_ext" git+https://github.com/NVIDIA/apex.git
ln -sf $CONDA_PREFIX/lib/python3.10/site-packages/nvidia/*/include/* $CONDA_PREFIX/include/
ln -sf $CONDA_PREFIX/lib/python3.10/site-packages/nvidia/*/include/* $CONDA_PREFIX/include/python3.10
pip install "transformer-engine[pytorch]==1.13.0"
```