LENS/pixi.toml at main · AdamDHines/LENS · GitHub

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
64
65
[project]
name = "lens"
version = "0.1.5"
description = "Neuromorphic event-based visual place recognition"
authors = ["Adam D Hines <adam.hines@qut.edu.au>", "Michael Milford <michael.milford@qut.edu.au", "Tobias Fischer <tobias.fischer@qut.edu.au>"]
channels = ["conda-forge"]
platforms = ["linux-64", "linux-aarch64", "osx-arm64", "osx-64"]
preview = ["pixi-build"]

[dependencies]
python = ">=3.6,<3.12"
pytorch = ">=2.1.1"
torchvision = ">=0.16.1"
numpy = ">=1.26.2,<2"
pandas = ">=2.1.1"
tqdm = ">=4.65.0"
prettytable = ">=3.5.0"
scikit-learn = ">=1.2.2"
sinabs = ">=3.0.1"
h5py = ">=3.10.0"
imageio = ">=2.34.1"
matplotlib-base = ">=3.8.2"
pynmea2 = ">=1.19.0"
scipy = ">=1.11.4"
scikit-image = ">=0.24.0"
wandb = ">=0.16.2"

[pypi-dependencies]
samna = ">=0.46.0.0"

[environments]
cuda = ["cuda"]

[feature.cuda]
platforms = ["linux-64"]
system-requirements = {cuda = "12"}

[feature.cuda.dependencies]
cuda-version = "12.*"
pytorch-gpu = "*"
cuda-cudart-dev = "*"
cuda-crt = "*"
cudnn = "*"
libcusparse-dev = "*"
cuda-driver-dev = "*"
cuda-nvcc = "*"
cuda-nvrtc-dev = "*"
cuda-nvtx-dev = "*"
cuda-nvml-dev = "*"
cuda-profiler-api = "*"
cusparselt = "*"
libcublas-dev = "*"
libcudss-dev = "*"
libcufile-dev = "*"
libcufft-dev = "*"
libcurand-dev = "*"
libcusolver-dev = "*"

[tasks]
demo = {cmd = "pixi run --frozen python main.py --matching --PR_curve --demo"}
train = {cmd = "pixi run python main.py --train_model"}
evaluate = {cmd = "pixi run python main.py --matching"}
optimizer = {cmd = "pixi run python optimizer.py"}
sim-speck = {cmd = "pixi run python main.py --simulated_speck --matching"}
on-speck = {cmd = "pixi run python main.py --event_driven"}