From ee6fae291d9a21dd3090e176297958e01afa8fa5 Mon Sep 17 00:00:00 2001 From: "Stephen R. Aylward" Date: Thu, 20 Aug 2026 11:50:31 -0400 Subject: [PATCH] Update installation.rst Recommend installing [all] option. Signed-off-by: Stephen R. Aylward --- docs/installation.rst | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) diff --git a/docs/installation.rst b/docs/installation.rst index 9f3efb9..a537ac7 100644 --- a/docs/installation.rst +++ b/docs/installation.rst @@ -35,16 +35,19 @@ Installation Methods Method 1: Install from PyPI (Recommended) ------------------------------------------ -Install the ``[cuda13]`` extra. It enables every feature and gives the best +Install the ``[all]`` extra. It enables every feature and gives the best performance: .. code-block:: bash - uv pip install "physiotwin4d[cuda13]" + uv pip install "physiotwin4d[all]" -The ``[cuda13]`` extra installs CuPy. In uv-managed source environments, -PyTorch, torchvision, and torchaudio resolve from the CUDA 13.0 PyTorch wheel -index. There is no need to install PyTorch separately. +The ``[all]`` extra installs PhysicsNeMo, CuPy, and dependencies for +development, testing, and documenting. In uv-managed source environments, +PyTorch, torchvision, and torchaudio should resolve from the CUDA 13.0 PyTorch wheel +index. However, that automation is not garaunteed for every platform, and it is recommended +to pre-install torch with CUDA acceleration, e.g., as described at +https://pytorch.org/get-started/locally/. CPU-only fallback (evaluation, or no NVIDIA GPU available):