aphrodite-engine/environment.yaml

name: aphrodite
channels:
  - pytorch
  - nvidia/label/cuda-11.8.0
  - conda-forge
  - defaults
dependencies:
  - python=3.10.*
  - pytorch=2.0.*
  - pytorch-cuda=11.8
  - cuda-nvcc=11.8
  - cuda-libraries-dev=11.8
  - cxx-compiler
  - pip
  - git=2.35.1
  - ninja