Skip to content

GitLab

Commit 64d88f4c authored by chenpangpang's avatar chenpangpang
Browse files

feat: 新增onnxruntime-gpu和einops依赖

parent aacfdabb
Pipeline #1422 canceled with stages
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 1 deletion
Dockerfile
View file @ 64d88f4c
FROM image.sourcefind.cn:5000/gpu/admin/base/jupyterlab-pytorch:2.2.0-python3.10-cuda12.1-ubuntu22.04 as base FROM image.sourcefind.cn:5000/gpu/admin/base/jupyterlab-pytorch:2.2.0-python3.10-cuda12.1-ubuntu22.04 as base
RUN cd /root && git clone -b gpu http://developer.hpccube.com/codes/chenpangpang/photomaker.git RUN cd /root && git clone -b gpu http://developer.hpccube.com/codes/chenpangpang/photomaker.git
WORKDIR /root/photomaker/PhotoMaker WORKDIR /root/photomaker/PhotoMaker
RUN pip install -r requirements.txt RUN apt-get update && apt-get install -y gcc g++
RUN pip install -r requirements.txt && \
pip install onnxruntime-gpu==1.18.0 --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/
RUN pip install git+https://github.com/TencentARC/PhotoMaker.git RUN pip install git+https://github.com/TencentARC/PhotoMaker.git
COPY chenyh/photomaker/TencentARC/PhotoMaker /root/photomaker/PhotoMaker/TencentARC/PhotoMaker COPY chenyh/photomaker/TencentARC/PhotoMaker /root/photomaker/PhotoMaker/TencentARC/PhotoMaker
......
PhotoMaker/requirements.txt
View file @ 64d88f4c
...@@ -11,3 +11,4 @@ omegaconf ...@@ -11,3 +11,4 @@ omegaconf
peft peft
gradio gradio
insightface==0.7.3 insightface==0.7.3
einops
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