Home Explore Help
Register Sign In
david
/
aphrodite-engine
mirror of https://github.com/PygmalionAI/aphrodite-engine
1
0
Fork 0
Files Issues 0 Wiki
Branch: dev
Branches Tags
aphrodite-engin...
/
requirements-cuda.txt

requirements-cuda.txt 188 B
Permalink History Raw

123456789 
  1. # Common dependencies
    2. 

  2. -r requirements-common.txt
    3. 

  3. 

  4. # Dependencies for NVIDIA GPUs
    5. 

  5. ray >= 2.9
    6. 

  6. pynvml == 11.5.0
    7. 

  7. torch == 2.3.0
    8. 

  8. xformers == 0.0.26.post1  # Requires torch 2.3.0
    9. 

  9. triton >= 2.2.0
    10.