Commit ac378b7c authored by zhuwenwen's avatar zhuwenwen
Browse files

update deps

parent 88b5fd38
FROM image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-centos7.6-dtk24.04-py310
ENV LANG C.UTF-8
RUN pip install ray==2.9.1 -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com
RUN pip install ray==2.9.1 tiktoken aiohttp==3.9.1 outlines==0.0.37 openai==1.23.3 -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com
\ No newline at end of file
......@@ -33,8 +33,8 @@ docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-centos7.6-dtk2
# <Container Path>容器映射路径
docker run -it --name qwen1.5_vllm --privileged --shm-size=64G --device=/dev/kfd --device=/dev/dri/ --cap-add=SYS_PTRACE --security-opt seccomp=unconfined --ulimit memlock=-1:-1 --ipc=host --network host --group-add video -v /opt/hyhal:/opt/hyhal -v <Host Path>:<Container Path> <Image ID> /bin/bash
# 更新镜像的ray版本
pip install ray==2.9.1
# 更新镜像的ray版本和服务依赖
pip install ray==2.9.1 tiktoken aiohttp==3.9.1 outlines==0.0.37 openai==1.23.3
```
### Dockerfile(方法二)
......@@ -48,6 +48,7 @@ docker run -it --name qwen1.5_vllm --privileged --shm-size=64G --device=/dev/kf
### Anaconda(方法三)
```
conda create -n qwen1.5_vllm python=3.10
pip install ray==2.9.1 tiktoken aiohttp==3.9.1 outlines==0.0.37 openai==1.23.3
```
关于本项目DCU显卡所需的特殊深度学习库可从[光合](https://developer.hpccube.com/tool/)开发者社区下载安装。
* DTK驱动:dtk24.04
......
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