Skip to content

GitLab

Commit 576f3940 authored by chenpangpang's avatar chenpangpang
Browse files

feat: 拉取v2-gpu分支

parent 5a2ee59a
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
Dockerfile
View file @ 576f3940
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 apt-get update && apt-get install -y gcc g++ RUN apt-get update && apt-get install -y gcc g++
RUN cd /root && git clone -b gpu http://developer.hpccube.com/codes/chenpangpang/photomaker.git RUN cd /root && git clone -b v2-gpu http://developer.hpccube.com/codes/chenpangpang/photomaker.git
WORKDIR /root/photomaker/PhotoMaker WORKDIR /root/photomaker/PhotoMaker
RUN pip install -r requirements.txt && \ 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/ pip install onnxruntime-gpu==1.18.0 --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/
......
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