environment.yaml 288 B

123456789101112131415161718
  1. name: aphrodite
  2. channels:
  3. - nvidia
  4. - pytorch
  5. - nvidia/label/cuda-12.1.1
  6. - conda-forge
  7. - defaults
  8. dependencies:
  9. - python=3.11.*
  10. - pytorch=2.2.*
  11. - pytorch-cuda=12.1
  12. - cuda-nvcc=12.1
  13. - cuda-libraries-dev=12.1
  14. - cxx-compiler
  15. - gxx=11.3
  16. - pip
  17. - git=2.35.1
  18. - ninja