-
Notifications
You must be signed in to change notification settings - Fork 0
/
pixi.toml
63 lines (54 loc) · 2.13 KB
/
pixi.toml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
[project]
authors = ["Christian Döring <[email protected]>"]
channels = ["nvidia/label/cuda-12.4.1", "nvidia", "conda-forge", "pytorch"]
description = "Add a short description here"
name = "neural_interference"
platforms = ["linux-64"]
version = "0.1.0"
[tasks]
test-mitsuba = { cmd = "echo $PYTHONPATH && python -m pytest", cwd = "build-mitsuba", env = { PYTHONPATH = "python:$PYTHONPATH" }, depends-on = [
"build-mitsuba",
] }
debug-mitsuba = { cmd = "echo $PYTHONPATH && gdb --args python -m pytest", cwd = "build-mitsuba", env = { PYTHONPATH = "python:$PYTHONPATH" }, depends-on = [
"build-mitsuba",
] }
build-mitsuba = { cwd = "build-mitsuba", cmd = "ninja", depends-on = [
"configure-mitsuba",
] }
configure-mitsuba = { cmd = "cmake -DCMAKE_EXPORT_COMPILE_COMMANDS=on -DCMAKE_BUILD_TYPE=RelWithDebInfo -G Ninja -S mitsuba3 -B build-mitsuba" }
clean-mitsuba = { cmd = "rm -r build", cwd = "mitsuba3" }
[dependencies]
python = "3.12.*"
gcc = { version = "13.2.*", channel = "conda-forge" }
gxx = { version = "13.2.*", channel = "conda-forge" }
pytorch-cuda = { version = "12.4.*", channel = "pytorch" }
pytorch = { version = "2.5.0", channel = "pytorch" }
torchvision = { version = "0.20.0", channel = "pytorch" }
torchaudio = { version = "2.5.0", channel = "pytorch" }
zlib = "1.3.1"
libcxx = ">=19.1.1,<20"
libcxx-devel = ">=19.1.1,<20"
libcxxabi = ">=19.1.1,<20"
ninja = ">=1.12.1,<2"
pip = ">=24.2,<25"
ipython = ">=8.28.0,<9"
matplotlib = ">=3.9.2,<4"
numpy = ">=2.1.2,<3"
omegaconf = ">=2.3.0,<3"
pytest = ">=8.3.3,<9"
[system-requirements]
cuda = "12.4"
[pypi-options]
no-build-isolation = ["tinycudann"]
[pypi-dependencies]
tensorboardx = "*"
mitsuba = ">=3.6.0, <4"
tinycudann = { git = "https://github.com/NVlabs/tiny-cuda-nn.git", subdirectory = "bindings/torch", rev = "c91138bcd4c6877c8d5e60e483c0581aafc70cce" }
[activation]
scripts = ["build-mitsuba/setpath.sh"]
# PYTHONPATH = "mitsuba3/build/python:$PYTHONPATH"
[target.linux-64.activation.env]
# Have to expose /lib64 for libcuda.so (cuda driver)
LIBRARY_PATH = "/usr/lib/x86_64-linux-gnu:/lib64:$LIBRARY_PATH"
CMAKE_PREFIX_PATH = "$CONDA_PREFIX"
LD_LIBRARY_PATH = "$CONDA_PREFIX/lib"