Commit 98ca21b7 authored by chenpangpang's avatar chenpangpang
Browse files

feat: 修改Dockerfile,run.sh保证在dcu上能运行

parent 8a92b74a
Pipeline #1363 failed with stages
in 0 seconds
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/dcu/admin/base/jupyterlab-pytorch:2.1.0-ubuntu20.04-dtk24.04.1-py3.10 as base
RUN cd /root && git clone -b dev http://developer.hpccube.com/codes/chenpangpang/kolors.git RUN cd /root && git clone -b dcu http://developer.hpccube.com/codes/chenpangpang/kolors.git
WORKDIR /root/kolors/Kolors WORKDIR /root/kolors/Kolors
RUN conda create --name kolors -y python=3.8 && source activate kolors && pip install -r requirements.txt && python3 setup.py install RUN pip install -r requirements.txt && python3 setup.py install
RUN rm -r build dist kolors.egg-info RUN rm -r build dist kolors.egg-info
######### #########
# Prod # # Prod #
######### #########
FROM image.sourcefind.cn:5000/gpu/admin/base/jupyterlab-pytorch:2.2.0-python3.10-cuda12.1-ubuntu22.04 FROM image.sourcefind.cn:5000/gpu/admin/base/jupyterlab-pytorch:2.2.0-python3.10-cuda12.1-ubuntu22.04
RUN conda create --name kolors python=3.8 COPY --from=base /opt/conda/lib/python3.10/site-packages /opt/conda/lib/python3.10/site-packages
COPY --from=base /opt/conda/envs/kolors/lib/python3.8/site-packages /opt/conda/envs/kolors/lib/python3.8/site-packages
COPY --from=base /root/kolors/Kolors /root/Kolors COPY --from=base /root/kolors/Kolors /root/Kolors
COPY --from=base /root/kolors/启动器.ipynb /root/kolors/run.sh /root/ COPY --from=base /root/kolors/启动器.ipynb /root/kolors/run.sh /root/
RUN mkdir /root/Kolors/weight RUN mkdir /root/Kolors/weight
......
#!/bin/bash #!/bin/bash
cd /root/Kolors cd /root/Kolors
source activate kolors
python3 scripts/sampleui.py python3 scripts/sampleui.py
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