ViPlan/environment.yml at main · merlerm/ViPlan · 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
66
67
68
69
70
71
72
name: viplan_env
channels:
  - nvidia
  - conda-forge
dependencies:
  - python=3.12
  - numpy
  - pandas
  - matplotlib
  - seaborn
  - jupyterlab
  - ipywidgets
  - ipykernel
  - transformers
  - datasets
  - huggingface_hub
  - llama-cpp-python
  # Compiler packages, must be supported by CUDA
  # https://docs.nvidia.com/cuda/cuda-installation-guide-linux/#host-compiler-support-policy
  - compilers
  - gcc>=11.4,<13.3
  - gxx>=11.4,<13.3
  - gfortran>=11.4,<13.3
  - clang>=7,<20.0
  # CUDA versions are determined by pytorch
  - cudnn>=8.9.2,<10
  - cuda-version>=12.6,<12.7
  - cuda-compiler>=12.6,<12.7
  - cuda-nvcc>=12.6,<12.7
  - cuda-libraries-dev>=12.6,<12.7
  - cmake
  - llama-index
  - llama-index-llms-openai
  - llama-index-readers-file
  - accelerate
  - python-dotenv
  - pillow-heif
  - faiss-gpu
  - flask
  - pip
  - pip:
      - torch
      - torchvision
      - torchaudio
      - --extra-index-url https://download.pytorch.org/whl/cu126
      - vllm==0.12.0
      - peft
      - bitsandbytes
      - pymupdf4llm
      - llama-index-llms-vllm
      - llama-index-llms-huggingface
      - llama-index-embeddings-huggingface
      - llama-index-llms-huggingface-api
      - mloggers
      - fasteners
      - fire
      - unified_planning
      - unified_planning[engines]
      - timm
      - backoff
      - anthropic
      - google-genai
      - tabulate
      # flash-attn must be installed manually with pip install flash-attn --no-build-isolation
      # transformers can also be downgraded to 4.50.3 with pip install transformers==4.50.3 to run Molmo
      # see https://huggingface.co/allenai/Molmo-7B-D-0924/discussions/44


variables:
  LLAMA_CUBLAS: 1
  FORCE_CMAKE: 1
  CMAKE_ARGS: "-DLLAMA_CUBLAS=on"