File size: 945 Bytes
a0d95b0 | 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 | # 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
|