mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-15 20:33:59 +08:00
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:
1 parent
50e0e3c70c
commit
c0c329a2f1
9 files changed
+38
-33
No files matched your search
@@ -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:
|
||||
|
||||
Reference in new issue
Block a user