Commit Graph

14 Commits

Author SHA1 Message Date
hoshi-hiyouga
5ad86fc71d Update Dockerfile
Former-commit-id: 5e8a619d1d
2024-11-02 20:20:26 +08:00
Valerio Mariani
1ab3ea21ce make base image parametric.
default `BASE_IMAGE` is nvcr.io/nvidia/pytorch:24.02-py3 for retro-compatibility


Former-commit-id: d5b2def7c0
2024-10-30 21:53:32 +01:00
StrangeBytesDev
d1e766898c Add additional install options to Dockerfiles
Former-commit-id: 237e302b5c
2024-09-24 16:54:46 -07:00
hzhaoy
93ba3bd5b0 fix #4780
Former-commit-id: 642c6d666f
2024-07-12 00:25:48 +08:00
hzhaoy
b3e4793ded fix #4779
Former-commit-id: a8bf1abf0f
2024-07-12 00:15:15 +08:00
hiyouga
cba994b9dc fix docker flashattn
Former-commit-id: e3141f5f1b
2024-06-28 01:28:59 +08:00
hiyouga
5e60c29971 fix docker file
Former-commit-id: 0f421055da
2024-06-27 20:29:16 +08:00
hiyouga
7c488cea57 tiny fix
Former-commit-id: e44a4f07f0
2024-06-27 20:14:48 +08:00
hzhaoy
c662c2e56f add flash-attn installation flag in Dockerfile
Former-commit-id: e19491b0f0
2024-06-27 00:13:30 +08:00
hzhaoy
08a221443c support flash-attn in Dockerfile
Former-commit-id: c88b1be9f3
2024-06-25 15:13:07 +08:00
hiyouga
dafc9268bc fix #4419
Former-commit-id: efb81b25ec
2024-06-25 01:51:29 +08:00
hoshi-hiyouga
cfa2dbefcb Update Dockerfile
Former-commit-id: fce146ab68
2024-06-25 00:46:08 +08:00
hoshi-hiyouga
f84bce3638 Update Dockerfile
Former-commit-id: dcc2e24f5c
2024-06-24 23:41:35 +08:00
MengqingCao
3b499948a5 update docker files
1. add docker-npu (Dockerfile and docker-compose.yml)
  2. move cuda docker to docker-cuda and tiny changes to adapt to the new path


Former-commit-id: d7207e8ad1
2024-06-24 10:57:36 +00:00