19 lines
1.1 KiB
Markdown
19 lines
1.1 KiB
Markdown
# 运行方式
|
|
|
|
```python
|
|
docker pull git.modelhub.org.cn:9443/enginex-hygon/pytorch:2.4.1-ubuntu22.04-dtk25.04-py3.10-fixpy
|
|
# 为以上拉取的docker的镜像ID替换
|
|
docker run -it --name mineru --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 $PWD/MinerU_pytorch:/home/MinerU_pytorch /bin/bash
|
|
cd /home/MinerU_pytorch
|
|
pip install -e .[core] -i https://mirrors.aliyun.com/pypi/simple/
|
|
pip install numpy==1.24.3
|
|
#以下torchvision-0.19.1、torch-2.4.1、triton-3.0.0whl可从光合开发者社区下载安装： https://developer.sourcefind.cn/tool/
|
|
pip install torchvision-0.19.1+das.opt2.dtk2504-cp310-cp310-manylinux_2_28_x86_64.whl
|
|
pip install torch-2.4.1+das.opt2.dtk2504-cp310-cp310-manylinux_2_28_x86_64.whl
|
|
pip install triton-3.0.0+das.opt4.dtk2504-cp310-cp310-manylinux_2_28_x86_64.whl
|
|
cd sglang-v0.4.6.post5.dev/sgl-kernel
|
|
python setup_hip.py install
|
|
cd ..
|
|
pip install -e "python[all_hip]"
|
|
```
|