OpenSceneFlow/environment.yaml at main · KTH-RPL/OpenSceneFlow · 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
name: opensf
channels:
  - conda-forge
  - pytorch
dependencies:
  - python=3.8
  - pytorch::pytorch=2.0.0
  - pytorch::torchvision
  - pytorch::pytorch-cuda=11.7
  - nvidia/label/cuda-11.7.0::cuda
  - lightning==2.0.1
  - mkl==2024.0.0
  - tensorboard
  - numba
  - numpy
  - pandas
  - pip
  - scipy
  - tqdm
  - h5py
  - wandb
  - omegaconf
  - hydra-core
  - fire
  - tabulate
  - scikit-learn==1.3.2
  - hdbscan
  - setuptools==69.5.1
  - gxx_linux-64==11.4.0
  - pathtools
  - rerun-sdk
  - pip:
    - assets/cuda/chamfer3D
    - assets/cuda/mmcv
    - open3d==0.18.0
    - av2==0.2.1
    - spconv-cu117==2.3.6

    # Qingwen's public pkg:
    - dztimer
    - dufomap==1.1.0
    - linefit==1.1.0

# Reason about the version fixed:
# setuptools==68.5.1: https://github.com/aws-neuron/aws-neuron-sdk/issues/893
# mkl==2024.0.0: https://github.com/pytorch/pytorch/issues/123097#issue-2218541307
# av2==0.2.1: in case other version deleted some functions.
# lightning==2.0.1: https://stackoverflow.com/questions/76647518/how-to-fix-error-cannot-import-name-modelmetaclass-from-pydantic-main
# open3d==0.18.0: because 0.17.0 have bug on set the view json file
# dufomap==1.0.0: in case later updating may not compatible with the code.
# spconv-cu117==2.3.6: avoid error: KeyError: ((16, 8, 8), float, float)
# torch-scatter==2.1.2: in case later updating may not compatible with the code.