Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
Mixtral_vllm
Commits
398fd91b
Commit
398fd91b
authored
Feb 10, 2025
by
laibao
Browse files
更新镜像,更新环境变量(包含绑卡顺序,同流推理)
parent
c55c86d9
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
15 additions
and
3 deletions
+15
-3
Dockerfile
Dockerfile
+1
-1
README.md
README.md
+14
-2
No files found.
Dockerfile
View file @
398fd91b
FROM
image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.3.0-
py3.10
-dtk24.04.3-
ubuntu20.04-vllm0.6
FROM
image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.3.0-
ubuntu22.04
-dtk24.04.3-
py3.10
ENV
LANG C.UTF-8
ENV
LANG C.UTF-8
README.md
View file @
398fd91b
...
@@ -38,7 +38,7 @@ Mixtral模型是一种稀疏专家混合(SMoE)语言模型,在每层包含
...
@@ -38,7 +38,7 @@ Mixtral模型是一种稀疏专家混合(SMoE)语言模型,在每层包含
提供
[
光源
](
https://www.sourcefind.cn/#/image/dcu/custom
)
拉取推理的docker镜像:
提供
[
光源
](
https://www.sourcefind.cn/#/image/dcu/custom
)
拉取推理的docker镜像:
```
```
docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.3.0-
py3.10
-dtk24.04.3-
ubuntu20.04-vllm0.6
docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.3.0-
ubuntu22.04
-dtk24.04.3-
py3.10
# <Image ID>用上面拉取docker镜像的ID替换
# <Image ID>用上面拉取docker镜像的ID替换
# <Host Path>主机端路径
# <Host Path>主机端路径
# <Container Path>容器映射路径
# <Container Path>容器映射路径
...
@@ -73,7 +73,19 @@ conda create -n mixtral_vllm python=3.10
...
@@ -73,7 +73,19 @@ conda create -n mixtral_vllm python=3.10
*
vllm: 0.6.2
*
vllm: 0.6.2
*
python: python3.10
*
python: python3.10
`Tips:需先安装相关依赖,最后安装vllm包`
`Tips:需先安装相关依赖,最后安装vllm包`
环境变量:
export ALLREDUCE_STREAM_WITH_COMPUTE=1
export VLLM_NUMA_BIND=1
export VLLM_RANK0_NUMA=0
export VLLM_RANK1_NUMA=1
export VLLM_RANK2_NUMA=2
export VLLM_RANK3_NUMA=3
export VLLM_RANK4_NUMA=4
export VLLM_RANK5_NUMA=5
export VLLM_RANK6_NUMA=6
export VLLM_RANK7_NUMA=7
## 数据集
## 数据集
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment