4
0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-11 02:24:38 +08:00

Adding explicit pytorch version for Cuda 11.7 installs (#1611)

Co-authored-by: Matthew Tancik <tancik@berkeley.edu>
This commit is contained in:
Julian Rendell 2023-03-15 20:56:34 -07:00 committed by GitHub
parent 4c2e200098
commit fac13d0bab
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -71,7 +71,7 @@ pip uninstall torch torchvision functorch
Install pytorch 1.13.1 with CUDA and [tiny-cuda-nn](https://github.com/NVlabs/tiny-cuda-nn)
```bash
pip install torch torchvision functorch --extra-index-url https://download.pytorch.org/whl/cu117
pip install torch==1.13.1 torchvision functorch --extra-index-url https://download.pytorch.org/whl/cu117
```
:::