Commit 202f39e7 authored by jerrrrry's avatar jerrrrry
Browse files

Update Dockerfile

parent b5a8c7f5
# 使用官方光源基础镜像
FROM vllm:085
# 使用官方光源基础镜像
FROM image.sourcefind.cn:5000/dcu/admin/base/custom:vllm0.8.5-ubuntu22.04-dtk25.04-rc7-das1.5-py3.10-20250521-fixpy-rocblas0521-beta2
# 创建目标目录
RUN mkdir -p /workspace/test/results
......@@ -11,7 +11,6 @@ WORKDIR /workspace/test/results
# 将主机上的 test.sh 复制到容器中的 /workspace/test
COPY ./run.sh /workspace/test
COPY ./benchmark_throughput_0.7.2.py /workspace/test
COPY ./topo.xml /workspace/test
COPY ./models-to-test.cfg /workspace/test
# 确保 test.sh 有可执行权限
......@@ -20,9 +19,23 @@ RUN chmod +x /workspace/test/run.sh
# 设置容器启动时运行的命令
# 使用单个 CMD 执行所有命令
CMD bash -c "\
rocm-bandwidth-test > rocm-bandwidth-test.txt && \
hy-smi > hy-smi.txt && \
hy-smi -c > hy-smi-c.txt && \
pip list > pip-list.txt && \
lscpu > lscpu.txt && \
bash /workspace/test/run.sh > test.log 2>&1"
\ No newline at end of file
rocm-bandwidth-test > rocm-bandwidth-test.log && \
hy-smi > hy-smi.log && \
hy-smi -c > hy-smi-c.log && \
pip list > pip-list.log && \
lscpu > lscpu.log && \
git clone https://www.ghproxy.cn/github.com/ROCm/rccl-tests.git --depth 1 -b master && \
cd /workspace/test/results/rccl-tests && \
source /opt/dtk/env.sh && \
make MPI=1 MPI_HOME=/opt/mpi ROCM_HOME=/opt/dtk/ NCCL_HOME=/opt/dtk/rccl/ CUSTOM_RCCL_LIB=/opt/dtk/rccl/lib/librccl.so -j32 && \
./build/all_reduce_perf -b 8 -e 1G -f 2 -g 8 > /workspace/test/results/all_reduce_perf_8.log && \
./build/all_reduce_perf -b 4 -e 1G -f 2 -g 4 > /workspace/test/results/all_reduce_perf_4.log && \
cd /workspace/test/results && \
git clone http://developer.sourcefind.cn/codes/OpenDAS/dcu_env_check.git && \
apt-get update && \
apt-get install -y iproute2 dmidecode ipmitool && \
cd /workspace/test/results/dcu_env_check && \
bash system_check.sh > /workspace/test/results/system_check.log && \
cd /workspace/test/results && \
bash /workspace/test/run.sh > test.log 2>&1"
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