tink/nerfstudio
4
0 Fork 0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-12 10:55:57 +08:00
Code Activity

Update docs to recommend torch 2.1 (#2748)

Browse Source
This commit is contained in:
Brent Yi authored and GitHub committed 2024-01-11 15:12:41 -08:00
1 parent 3b6ca91dbc
commit 55b7a2c8ef
2 files changed
+4 -13
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

docs/quickstart/installation.md
+3 -3
View File
@@ -42,12 +42,12 @@ pip uninstall torch torchvision functorch tinycudann
```
::::{tab-set}
:::{tab-item} Torch 2.0.1 with CUDA 11.8
:::{tab-item} Torch 2.1.2 with CUDA 11.8 (recommended)
Install PyTorch 2.0.1 with CUDA 11.8:
Install PyTorch 2.1.2 with CUDA 11.8:
```bash
pip install torch==2.0.1+cu118 torchvision==0.15.2+cu118 --extra-index-url https://download.pytorch.org/whl/cu118
pip install torch==2.1.2+cu118 torchvision==0.16.2+cu118 --extra-index-url https://download.pytorch.org/whl/cu118
```
To build the necessary CUDA extensions, `cuda-toolkit` is also required. We