14 lines
741 B
Markdown
14 lines
741 B
Markdown
# 运行方式
|
|
|
|
```python
|
|
docker pull git.modelhub.org.cn:9443/enginex-hygon/pytorch:2.4.1-ubuntu22.04-dtk25.04.1-py3.10
|
|
docker run -it --name {name} --shm-size=1024G --device=/dev/kfd --device=/dev/dri/ --privileged --cap-add=SYS_PTRACE --security-opt seccomp=unconfined --ulimit memlock=-1:-1 --ipc=host --network host --group-add video -v /opt/hyhal:/opt/hyhal:ro -v {}:{} {docker_image} /bin/bash
|
|
# 修改1 {name} 需要改为自定义名称
|
|
# 修改2 {docker_image} 需要需要创建容器的对应镜像名称
|
|
# 修改3 -v 挂载路径到容器指定路径
|
|
pip install -r requirements.txt
|
|
cd LLaMA-Factory
|
|
pip install -e ".[torch,metrics]"
|
|
pip install deepspeed-0.14.2+das.opt1.dtk25041-cp310-cp310-manylinux_2_28_x86_64.whl
|
|
```
|