feat: dcu适配

4e7af75c · chenpangpang · 146f30d4 · 4e7af75c
Commit 4e7af75c authored Aug 09, 2024 by chenpangpang
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 3 deletions

Dockerfile Dockerfile +3 -3

No files found.
--- a/Dockerfile
+++ b/Dockerfile
-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 gpu https://developer.hpccube.com/codes/chenpangpang/tile-upscaler
+RUN cd /root && git clone -b dcu https://developer.hpccube.com/codes/chenpangpang/tile-upscaler
 WORKDIR /root/tile-upscaler/Tile-Upscaler
 RUN pip install -r requirements.txt
 COPY chenyh/tile-upscaler/models /root/tile-upscaler/Tile-Upscaler/models
@@ -8,7 +8,7 @@ COPY chenyh/tile-upscaler/CompVis /root/tile-upscaler/Tile-Upscaler/CompVis
 #########
 # 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/dcu/admin/base/jupyterlab-pytorch:2.1.0-ubuntu20.04-dtk24.04.1-py3.10
 RUN apt-get update && apt-get install -y libglib2.0-0 libgl1-mesa-glx
 COPY chenyh/tile-upscaler/frpc_linux_amd64_v0.2 /opt/conda/lib/python3.10/site-packages/gradio/
 RUN chmod +x /opt/conda/lib/python3.10/site-packages/gradio/frpc_linux_amd64_v0.2