| # environment.yml | |
| name: allm # The name of the environment | |
| channels: # The conda channels to search for packages | |
| # - pytorch | |
| - conda-forge | |
| # - dnachun | |
| # - anaconda | |
| channel_priority: strict | |
| dependencies: | |
| # Packages to install with conda | |
| - python=3.11.3 | |
| #- pytorch-cuda=12.4 | |
| #- pytorch | |
| # - numpy | |
| - transformers>=4.55 | |
| - einops | |
| - jaxtyping | |
| - tensorboard | |
| - omegaconf | |
| - accelerate | |
| - peft | |
| - wandb | |
| - scipy | |
| - pandas | |
| - matplotlib | |
| - scikit-image | |
| - scikit-learn | |
| - joblib | |
| - pillow | |
| - datasets | |
| ## NO - huggingface_hub | |
| - tqdm | |
| - nltk | |
| - future | |
| - defusedxml | |
| - ipdb | |
| - torchinfo | |
| - timm | |
| - graphviz #anaconda::graphviz | |
| - dnachun::torchviz | |
| - pip: | |
| # - draccus | |
| - fraction | |
| - vllm | |