environment.yml

# conda env create -f environment.yml
name: zobench
channels:
  - defaults
  - conda-forge
  - pytorch
  - huggingface
  - numba
  - nvidia
dependencies:
  - python=3.9.6
  - accelerate=0.25.0
  - aiohttp=3.9.1
  - aiosignal=1.3.1
  - appdirs=1.4.4
  - async-timeout=4.0.3
  - attrs=23.1.0
  - certifi=2023.11.17
  - charset-normalizer=3.3.2
  - click=8.1.7
  - datasets=2.16.0
  - dill=0.3.7
  - docker-pycreds=0.4.0
  - filelock=3.13.1
  - fsspec=2023.10.0
  - gitdb=4.0.11
  - GitPython=3.1.40
  - gmpy2=2.1.2
  - huggingface-hub=0.20.1
  - idna=3.6
  - Jinja2=3.1.2
  - joblib=1.3.2
  - llvmlite=0.41.1
  - MarkupSafe=2.1.3
  - mpmath=1.3.0
  - multidict=6.0.4
  - multiprocess=0.70.15
  - networkx=3.2.1
  - numba=0.58.1
  - numpy=1.26.2
  - packaging=23.2
  - pandas=2.1.4
  - pip=23.3.2
  - protobuf=4.25.1
  - psutil=5.9.7
  - pyarrow=14.0.2
  - pyarrow-hotfix=0.6
  - python-dateutil=2.8.2
  - pytz=2023.3.post1
  - PyYAML=6.0.1
  - regex=2023.12.25
  - requests=2.31.0
  - safetensors=0.4.1
  - scikit-learn=1.3.2
  - scipy=1.11.4
  - sentry-sdk=1.39.1
  - setproctitle=1.3.3
  - setuptools=68.2.2
  - six=1.16.0
  - smmap=5.0.1
  - sympy=1.12
  - threadpoolctl=3.2.0
  - tokenizers=0.13.3
  - torch=2.1.0
  - tqdm=4.66.1
  - transformers=4.28.1
  - typing_extensions=4.9.0
  - tzdata=2023.3
  - urllib3=2.1.0
  - wandb=0.16.1
  - wheel=0.42.0
  - xxhash=3.4.1
  - yarl=1.9.4