Merge pull request #2 from KAIR-BAIR/tancik/0.0.3

Update for pypi

examples/requirements.txt

+git+https://github.com/NVlabs/tiny-cuda-nn/#subdirectory=bindings/torch
+opencv-python
+imageio
+numpy
+tqdm
\ No newline at end of file

nerfacc/cuda/_backend.py

@@ -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(
-    name="nerfacc_cuda",
-    sources=sources,
-    extra_cflags=extra_cflags,
-    extra_cuda_cflags=extra_cuda_cflags,
-)
+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"]

pyproject.toml

@@ -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