Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
nerfacc
Commits
59f31c61
Commit
59f31c61
authored
Sep 14, 2022
by
Matt Tancik
Browse files
Update for pypi
parent
776dff86
Changes
3
Show whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
19 additions
and
16 deletions
+19
-16
examples/requirements.txt
examples/requirements.txt
+5
-0
nerfacc/cuda/_backend.py
nerfacc/cuda/_backend.py
+12
-6
pyproject.toml
pyproject.toml
+2
-10
No files found.
examples/requirements.txt
0 → 100644
View file @
59f31c61
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
View file @
59f31c61
...
@@ -2,6 +2,9 @@
...
@@ -2,6 +2,9 @@
import
glob
import
glob
import
os
import
os
from
subprocess
import
DEVNULL
,
call
from
subprocess
import
DEVNULL
,
call
from
rich.console
import
Console
console
=
Console
()
from
torch.utils.cpp_extension
import
load
from
torch.utils.cpp_extension
import
load
...
@@ -25,11 +28,14 @@ else:
...
@@ -25,11 +28,14 @@ else:
extra_cflags
=
[
"-O3"
]
extra_cflags
=
[
"-O3"
]
extra_cuda_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"
,
name
=
"nerfacc_cuda"
,
sources
=
sources
,
sources
=
sources
,
extra_cflags
=
extra_cflags
,
extra_cflags
=
extra_cflags
,
extra_cuda_cflags
=
extra_cuda_cflags
,
extra_cuda_cflags
=
extra_cuda_cflags
,
)
)
__all__
=
[
"_C"
]
__all__
=
[
"_C"
]
pyproject.toml
View file @
59f31c61
...
@@ -11,18 +11,10 @@ requires-python = ">=3.8"
...
@@ -11,18 +11,10 @@ requires-python = ">=3.8"
dependencies
=
[
dependencies
=
[
"ninja>=1.10.2.3"
,
"ninja>=1.10.2.3"
,
"pybind11>=2.10.0"
,
"pybind11>=2.10.0"
,
"torch>=1.12.1"
"torch>=1.12.1"
,
"rich>=12"
]
]
[project.optional-dependencies]
[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
# for documentation
docs
=
[]
docs
=
[]
\ No newline at end of file
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment