"docs/vscode:/vscode.git/clone" did not exist on "09daa438f8a1a8135e7b9960a3ab141dde70f85b"
Commit 59f31c61 authored by Matt Tancik's avatar Matt Tancik
Browse files

Update for pypi

parent 776dff86
git+https://github.com/NVlabs/tiny-cuda-nn/#subdirectory=bindings/torch
opencv-python
imageio
numpy
tqdm
\ No newline at end of file
......@@ -2,6 +2,9 @@
import glob
import os
from subprocess import DEVNULL, call
from rich.console import Console
console = Console()
from torch.utils.cpp_extension import load
......@@ -25,11 +28,14 @@ else:
extra_cflags = ["-O3"]
extra_cuda_cflags = ["-O3"]
_C = load(
with console.status(
"[bold yellow]Setting up CUDA (This may take a few minutes the first time)", spinner="bouncingBall"
):
_C = load(
name="nerfacc_cuda",
sources=sources,
extra_cflags=extra_cflags,
extra_cuda_cflags=extra_cuda_cflags,
)
)
__all__ = ["_C"]
......@@ -11,18 +11,10 @@ requires-python = ">=3.8"
dependencies = [
"ninja>=1.10.2.3",
"pybind11>=2.10.0",
"torch>=1.12.1"
"torch>=1.12.1",
"rich>=12"
]
[project.optional-dependencies]
# for example scripts
examples = [
"tinycudann @ git+https://github.com/NVlabs/tiny-cuda-nn/#subdirectory=bindings/torch",
"opencv-python",
"imageio",
"numpy",
"tqdm",
]
# for documentation
docs = []
\ No newline at end of file
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment