"vscode:/vscode.git/clone" did not exist on "303fede056660c168b101f54f69e4912e86da985"
Commit bd0fd267 authored by zhougaofeng's avatar zhougaofeng
Browse files

Update README.md

parent 3be6a7c3
...@@ -20,7 +20,7 @@ Yi-1.5-6B 是一个 decoder-only 的 transformer 模型,使用 SwiGLU激活函 ...@@ -20,7 +20,7 @@ Yi-1.5-6B 是一个 decoder-only 的 transformer 模型,使用 SwiGLU激活函
docker pull xxx docker pull xxx
docker run -it --shm-size=1024G -v /parastor/home/Yi-1.5-pytorch:/home/Yi-1.5-pytorch -v /opt/hyhal:/opt/hyhal --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name Yi-1.5 <your IMAGE ID> bash # <your IMAGE ID>为以上拉取的docker的镜像ID替换,本镜像为:c85ed27005f2 docker run -it --shm-size=1024G -v /parastor/home/Yi-1.5-pytorch:/home/Yi-1.5-pytorch -v /opt/hyhal:/opt/hyhal --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name Yi-1.5 <your IMAGE ID> bash # <your IMAGE ID>为以上拉取的docker的镜像ID替换,本镜像为:c85ed27005f2
cd /home/Yi-1.5-pytorch cd /home/Yi-1.5-pytorch
pip install -e .[torch,metrics] pip install -r requirement.txt -i https://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com
# deepspeed、bitsandbytes可从whl.zip文件里获取安装: # deepspeed、bitsandbytes可从whl.zip文件里获取安装:
pip install deepspeed-0.12.3+das1.0+gita724046.abi0.dtk2404.torch2.1.0-cp310-cp310-manylinux2014_x86_64.whl pip install deepspeed-0.12.3+das1.0+gita724046.abi0.dtk2404.torch2.1.0-cp310-cp310-manylinux2014_x86_64.whl
pip install bitsandbytes-0.42.0-py3-none-any.whl pip install bitsandbytes-0.42.0-py3-none-any.whl
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment