Skip to content

GitLab

Commit be7eec6d authored by chenpangpang's avatar chenpangpang
Browse files

feat: 修改dcu的onnxruntime的版本为dcu版本

parent 4c8da356
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
Dockerfile
View file @ be7eec6d
...@@ -6,7 +6,8 @@ COPY chenyh/InstantID/antelopev2 /root/instantid/InstantID/models/antelopev2 ...@@ -6,7 +6,8 @@ COPY chenyh/InstantID/antelopev2 /root/instantid/InstantID/models/antelopev2
COPY chenyh/InstantID/wangqixun /root/instantid/InstantID/wangqixun COPY chenyh/InstantID/wangqixun /root/instantid/InstantID/wangqixun
RUN apt-get update && apt-get install -y gcc g++ RUN apt-get update && apt-get install -y gcc g++
RUN pip install -r gradio_demo/requirements.txt && \ RUN pip install -r gradio_demo/requirements.txt && \
pip install onnxruntime-gpu==1.18.0 --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/ wget https://cancon.hpccube.com:65024/directlink/4/onnxruntime/DAS1.1/onnxruntime-1.15.0+das1.1.git739f24d.abi1.dtk2404-cp310-cp310-manylinux_2_31_x86_64.whl && \
pip install onnxruntime-1.15.0+das1.1.git739f24d.abi1.dtk2404-cp310-cp310-manylinux_2_31_x86_64.whl
######### #########
# Prod # # Prod #
......
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