requirements_GPU.txt 2.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117
  1. accelerate==1.15.0
  2. alembic==1.20.0
  3. annotated-types==0.8.0
  4. antlr4-python3-runtime==4.9.3
  5. anyio==4.15.1
  6. bcrypt==4.3.0
  7. Bottleneck==1.4.2
  8. brotlicffi==1.0.9.2
  9. certifi==2026.7.22
  10. cffi==1.17.1
  11. charset-normalizer==3.4.7
  12. click==8.5.0
  13. colorlog==6.10.1
  14. cryptography==45.0.5
  15. deepwind-wpf==0.1.0
  16. docker-pycreds==0.4.0
  17. einops==0.8.1
  18. ephem==4.2.1
  19. filelock==4.0.0
  20. flatbuffers==25.12.19
  21. formulaic==1.2.2
  22. fsspec==2026.7.0
  23. gitdb==4.0.12
  24. GitPython==3.2.0
  25. greenlet==3.1.1
  26. h11==0.16.0
  27. h5py==3.14.0
  28. hf-xet==1.6.0
  29. httpcore==1.0.9
  30. httpx==0.28.1
  31. huggingface_hub==0.36.2
  32. hydra-colorlog==1.2.0
  33. hydra-core==1.3.7
  34. idna==3.18
  35. importlib_metadata==9.0.1
  36. interface_meta==1.3.0
  37. invoke==3.0.3
  38. jaxtyping==0.3.11
  39. Jinja2==3.1.6
  40. joblib==1.5.3
  41. llvmlite==0.49.0
  42. Mako==1.4.1
  43. MarkupSafe==3.0.3
  44. mirrorearth_sdk==0.1.0
  45. mkl_fft==1.3.11
  46. mkl_random==1.2.8
  47. mkl-service==2.4.0
  48. mpi4py==4.1.2
  49. mpmath==1.3.0
  50. narwhals==2.26.0
  51. networkx==3.6.1
  52. numba==0.67.0
  53. numexpr==2.11.0
  54. numpy==1.26.4
  55. nvidia-cublas-cu12==12.4.5.8
  56. nvidia-cuda-cupti-cu12==12.4.127
  57. nvidia-cuda-nvrtc-cu12==12.4.127
  58. nvidia-cuda-runtime-cu12==12.4.127
  59. nvidia-cudnn-cu12==9.1.0.70
  60. nvidia-cufft-cu12==11.2.1.3
  61. nvidia-curand-cu12==10.3.5.147
  62. nvidia-cusolver-cu12==11.6.1.9
  63. nvidia-cusparse-cu12==12.3.1.170
  64. nvidia-cusparselt-cu12==0.6.2
  65. nvidia-nccl-cu12==2.21.5
  66. nvidia-nvjitlink-cu12==12.4.127
  67. nvidia-nvtx-cu12==12.4.127
  68. omegaconf==2.3.1
  69. optuna==4.6.0
  70. packaging==26.3
  71. pandas==2.3.2
  72. paramiko==5.0.0
  73. patsy==1.0.3
  74. peft==0.21.2
  75. pillow==10.4.0
  76. platformdirs==4.12.4
  77. protobuf==6.33.6
  78. psutil==7.2.2
  79. psycopg2-binary==2.9.13
  80. pvlib==0.15.2
  81. pycparser==3.0
  82. pydantic==2.14.0
  83. pydantic_core==2.50.0
  84. PyNaCl==1.5.0
  85. PySocks==1.7.1
  86. python-dateutil==2.9.0.post0
  87. pytz==2026.3.post1
  88. PyYAML==6.0.2
  89. regex==2026.9.29
  90. requests==2.34.2
  91. safetensors==0.8.0
  92. scikit-learn==1.7.1
  93. scipy==1.16.0
  94. sentry-sdk==2.71.0
  95. setproctitle==1.3.8
  96. six==1.17.0
  97. smmap==5.0.3
  98. SQLAlchemy==2.0.43
  99. statsmodels==0.15.0
  100. sympy==1.13.1
  101. threadpoolctl==3.5.0
  102. timesfm==3.0.2
  103. tokenizers==0.22.2
  104. torch==2.6.0+cu124
  105. torchaudio==2.6.0+cu124
  106. torchvision==0.21.0+cu124
  107. tqdm==4.70.0
  108. transformers==4.57.6
  109. triton==3.2.0
  110. typing_extensions==4.16.0
  111. typing-inspection==0.4.4
  112. tzdata==2026.3
  113. urllib3==2.5.0
  114. wadler_lindig==0.1.7
  115. wandb==0.19.11
  116. wrapt==2.4.0
  117. zipp==4.1.0