Commit Graph

25 Commits

Author SHA1 Message Date
Yineng Zhang
141a459644 fix: only upgrade nccl for cu128 (#5986) 2025-05-02 11:31:29 -07:00
Yineng Zhang
f4c191a712 chore: update Dockerfile (#5894) 2025-04-29 12:55:13 -07:00
lukec
417b44eba8 [Feat] upgrade pytorch2.6 (#5417) 2025-04-20 16:06:34 -07:00
Kebe
4a893d142d Refactor Dockerfile: unify CUDA logic and reduce image size by ~2.6 GB (#3749)
Signed-off-by: Kebe <mail@kebe7jun.com>
2025-03-08 03:01:13 -08:00
Shenggui Li
c9565e49e7 [docker] added rdma support (#3619) 2025-02-17 15:36:16 +08:00
Yineng Zhang
ac963be234 update flashinfer-python (#3557) 2025-02-14 09:52:56 +08:00
Yineng Zhang
fa1b40e00d use nvcr.io/nvidia/tritonserver:24.04-py3-min as base image (#3457) 2025-02-10 13:52:33 +08:00
Yineng Zhang
6186a8f889 update flashinfer install index url (#3293) 2025-02-05 00:44:35 +08:00
Yineng Zhang
b49d6d0fee support 12.5 CUDA runtime (#3231) 2025-01-31 20:31:38 +08:00
Yineng Zhang
cf142b6eb8 fix: update Dockerfile for cu118 (#3181) 2025-01-27 23:46:44 +08:00
Yineng Zhang
4e1e3cff20 fix #2528 (#2541) 2024-12-22 00:14:41 +08:00
Yineng Zhang
fae4e5e99a chore: bump v0.3.6.post3 (#2259) 2024-11-30 01:41:16 +08:00
Lianmin Zheng
39bb49d156 Update dockerfile to include datamodel_code_generator (#1492) 2024-09-22 04:49:16 -07:00
William Arnold
58cafe23a7 Add libibverbs-dev to Dockerfile (#1427) 2024-09-15 15:40:31 +09:00
Yineng Zhang
6aa8ad14f8 fix: resolve Python.h header missing (#1119) 2024-08-16 15:46:43 +10:00
Yineng Zhang
0076f11541 fix: use devel for Triton's compiler requirements (#1074) 2024-08-13 04:08:43 -07:00
Yineng Zhang
cb99ba4fc6 feat: update Dockerfile (#1033)
Co-authored-by: vhain <vhain6512@gmail.com>
2024-08-12 16:24:06 +10:00
Yineng Zhang
c31f084c71 chore: update vllm to 0.5.4 (#966) 2024-08-07 21:15:41 +10:00
Ying Sheng
001b0bdd08 Update the base image of the docker (#900) 2024-08-02 21:54:57 -07:00
Yineng Zhang
8628ab9c8b feat: add docker workflow (#751) 2024-07-27 03:54:51 +10:00
Ying Sheng
7802df1e2b Update readme 2024-07-25 08:45:06 -07:00
Liangsheng Yin
04ec6ba2ac Fix dockerfile and triton cache manager (#720) 2024-07-25 03:04:21 -07:00
Liangsheng Yin
8832ecb1e4 Reduce docker size (#632) 2024-07-16 16:12:12 -07:00
Ying Sheng
f6b29f6920 Update docker file (#629) 2024-07-16 01:12:37 -07:00
Ying Sheng
ac11388756 Add docker file (#588)
Co-authored-by: Ying Sheng <ying.sheng@databricks.com>
2024-07-04 00:53:49 -07:00