11 lines
480 B
Markdown
11 lines
480 B
Markdown
|
# 运行方式
|
||
|
|
|
||
|
|
```python
|
||
|
|
docker pull git.modelhub.org.cn:9443/enginex-hygon/pytorch:2.1.0-py3.10-dtk24.04.3-ubuntu20.04
|
||
|
|
docker run -it --shm-size=1024G -v : -v /opt/hyhal:/opt/hyhal --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name Yi-1.5 bash # 为以上拉取的docker的镜像ID替换
|
||
|
|
cd /home/Yi-1.5-pytorch
|
||
|
|
pip install -r requirements.txt -i https://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com
|
||
|
|
|
||
|
|
pip uninstall vllm
|
||
|
|
```
|