| # environment.yml | |
| name: iba # 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 >= 2.6 | |
| # - numpy | |
| # - tensorboard | |
| # - omegaconf | |
| - wandb | |
| - scipy | |
| - pandas | |
| - matplotlib | |
| - scikit-image | |
| - scikit-learn | |
| - joblib | |
| - pillow | |
| ## NO - huggingface_hub | |
| - tqdm | |
| - nltk | |
| # - future | |
| # - defusedxml | |
| # - ipdb | |
| # - torchinfo | |
| # - timm | |
| # - graphviz #anaconda::graphviz | |
| # - dnachun::torchviz | |
| - pip: | |
| - --index-url https://download.pytorch.org/whl/cu126 | |
| - torch | |
| - torchvision | |
| - torchaudio | |
| - --index-url https://pypi.org/simple | |
| - transformers>=4.55 | |
| - accelerate | |
| - einops | |
| - jaxtyping | |
| - peft | |
| - datasets | |
| # - fraction | |
| - draccus | |
| - vllm | |