16 lines
536 B
Markdown
16 lines
536 B
Markdown
# 运行方式
|
||
```python
|
||
拉取镜像：
|
||
docker pull git.modelhub.org.cn:9443/enginex-hygon/pytorch:2.1.0-ubuntu20.04-dtk24.04.1-py3.10
|
||
创建并启动容器：
|
||
docker run --shm-size 64g --network=host --name=llama_fastchat --privileged --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v /opt/hyhal:/opt/hyhal:ro -v : -it bash
|
||
|
||
cp -r mpirun/* ./
|
||
cd FastChat-main
|
||
pip3 install -e .
|
||
cd ../transformers-main
|
||
pip3 install -e .
|
||
pip3 uninstall wandb
|
||
pip3 install mpi4py
|
||
cd ..
|
||
``` |