Add support for NVIDIA Blackwell GPUs (sm_120, RTX 50-series)

- Bump gsplat to >=1.5.3 (sm_120-capable release line) and nerfacc to >=0.5.3
- Drop EOL Python 3.8/3.9: requires-python >=3.10, CI publish/doc workflows
- torch.load(..., weights_only=False) for torch>=2.6 checkpoint compatibility
- Use torch.amp custom_fwd/custom_bwd with device_type="cuda" (torch>=2.4 API)
- Replace deprecated PIL private encoder API in pil_to_numpy with np.asarray
  (fixes Pillow>=11 compatibility)
- Docs: add Torch 2.8+ / CUDA 12.8 install tab for Blackwell GPUs

Tested on RTX 5060 (sm_120) with torch 2.11.0+cu128: splatfacto and
nerfacto train end-to-end and ns-render produces correct output.
This commit is contained in:
yesMohanHere committed 2026-07-17 15:53:36 +05:30
1 parent 50e0e3c70c
commit c0c329a2f1
9 files changed
+38 -33

No files matched your search

+25 -3
View File
@@ -5,7 +5,7 @@
::::::{tab-set}
:::::{tab-item} Linux
Nerfstudio requires `python >= 3.8`. We recommend using conda to manage dependencies. Make sure to install [Conda](https://docs.conda.io/en/latest/miniconda.html) before proceeding.
Nerfstudio requires `python >= 3.10`. We recommend using conda to manage dependencies. Make sure to install [Conda](https://docs.conda.io/en/latest/miniconda.html) before proceeding.
:::::
:::::{tab-item} Windows
@@ -51,7 +51,7 @@ For example:
When updating, or if you close your terminal before you finish the installation and run your first `splatfacto`, you have to re-do this environment activation step.
:::
Nerfstudio requires `python >= 3.8`. We recommend using conda to manage dependencies. Make sure to install [Conda](https://docs.conda.io/en/latest/miniconda.html) before proceeding.
Nerfstudio requires `python >= 3.10`. We recommend using conda to manage dependencies. Make sure to install [Conda](https://docs.conda.io/en/latest/miniconda.html) before proceeding.
:::::
::::::
@@ -59,7 +59,7 @@ Nerfstudio requires `python >= 3.8`. We recommend using conda to manage dependen
## Create environment
```bash
conda create --name nerfstudio -y python=3.8
conda create --name nerfstudio -y python=3.10
conda activate nerfstudio
python -m pip install --upgrade pip
@@ -79,6 +79,28 @@ pip uninstall torch torchvision functorch tinycudann
```
::::{tab-set}
:::{tab-item} Torch 2.8+ with CUDA 12.8 (Blackwell / RTX 50-series)
Required for NVIDIA Blackwell GPUs (RTX 50-series, compute capability sm_120), which need CUDA 12.8 or newer:
```bash
pip install torch torchvision --index-url https://download.pytorch.org/whl/cu128
```
To build the necessary CUDA extensions, `cuda-toolkit` 12.8 is also required. We
recommend installing with conda:
```bash
conda install -c "nvidia/label/cuda-12.8.0" cuda-toolkit
```
When building CUDA extensions (gsplat, tiny-cuda-nn, ...) for a Blackwell GPU, make sure the architecture list includes sm_120:
```bash
export TORCH_CUDA_ARCH_LIST="12.0+PTX"
```
:::
:::{tab-item} Torch 2.1.2 with CUDA 11.8 (recommended)
Install PyTorch 2.1.2 with CUDA 11.8: