"src/include/common.cuh" did not exist on "5d2cafcb24097f86d33f7c5243a3c0f3800854ec"
Commit be7eec6d authored by chenpangpang's avatar chenpangpang
Browse files

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

parent 4c8da356
......@@ -6,7 +6,8 @@ COPY chenyh/InstantID/antelopev2 /root/instantid/InstantID/models/antelopev2
COPY chenyh/InstantID/wangqixun /root/instantid/InstantID/wangqixun
RUN apt-get update && apt-get install -y gcc g++
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 #
......
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