Commit Graph

13 Commits

Author SHA1 Message Date
leo-pony
acc52e0fe7 [npu] update cann base image and torch 2.4 (#7061)
* Update base npu container image version:The Python version required for Hugging Face Transformers is >= python3.10

* Fix the bug: arg type of INSTALL_DEEPSPEED shoud been string now.

* Update Ascend CANN, CANN-Kernel and corresponding torch and torch-npu version

* Upgrade torch-npu needs packages' version: torch==2.1.0 and torch-npu==2.4.0.post2
2025-02-25 23:32:01 +08:00
hiyouga
cd14336506 update wechat 2025-01-04 07:59:57 +00:00
shibingli@yeah.net
f1d76786e0 Add ARG HTTP_PROXY in Dockerfile to support HTTP proxy during image building. 2024-12-27 18:31:14 +08:00
MengqingCao
b3f4acd1b4 update npu base image 2024-08-21 09:12:38 +00:00
hiyouga
93e6fbb37d update npu docker 2024-06-30 21:05:31 +08:00
hiyouga
e44a4f07f0 tiny fix 2024-06-27 20:14:48 +08:00
hoshi-hiyouga
5fcd33fd4f Update Dockerfile 2024-06-27 19:57:40 +08:00
hoshi-hiyouga
9624af1df3 Update Dockerfile 2024-06-27 19:51:25 +08:00
fanjunliang
bdda0827b3 support docker-npu-[amd64|arm64] build 2024-06-27 15:25:12 +08:00
fanjunliang
8096f94a7d fix torch-npu dependency 2024-06-26 18:21:42 +08:00
hiyouga
efb81b25ec fix #4419 2024-06-25 01:51:29 +08:00
hoshi-hiyouga
3af936a76d Update Dockerfile 2024-06-25 00:50:34 +08:00
MengqingCao
d7207e8ad1 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
2024-06-24 10:57:36 +00:00