name: LION
channels:
  - pytorch
  - nvidia
  - conda-forge
  - defaults
  - open3d-admin
  - comet_ml
dependencies:
  - pip
  
  - python
  - pytorch
  - torchvision
  - cudatoolkit
  - matplotlib
  - tqdm
  - trimesh
  - scipy
  - scikit-learn

  - h5py
  - wandb
  - pyyaml
  - open3d
  - cupy
  - scikit-image
  - loguru
  - einops
  - comet_ml
  - diffusers
  - pip:
    - pykeops
    - nestargs
    - flatten_dict
    - point-cloud-utils
    - calmsize
    # - git+https://github.com/openai/CLIP.git