Skip to content

GitLab

Commit 8c68a142 authored by chenpangpang's avatar chenpangpang
Browse files

feat: 更新依赖版本

parent ea2fb69d
Pipeline #2161 canceled with stages
Hide whitespace changes
Inline Side-by-side
Showing with 18 additions and 14 deletions
Dockerfile
View file @ 8c68a142
FROM image.sourcefind.cn:5000/gpu/admin/base/jupyterlab-pytorch:2.3.1-py3.10-cuda11.8-ubuntu22.04-devel as base FROM image.sourcefind.cn:5000/gpu/admin/base/jupyterlab-pytorch:2.1.2-py3.10-cuda12.1-ubuntu22.04-devel as base
ARG IMAGE=nvcomposer ARG IMAGE=nvcomposer
ARG IMAGE_UPPER=NVComposer ARG IMAGE_UPPER=NVComposer
ARG BRANCH=gpu ARG BRANCH=gpu
...@@ -10,7 +10,7 @@ RUN pip install -r requirements.txt ...@@ -10,7 +10,7 @@ RUN pip install -r requirements.txt
######### #########
# Prod # # Prod #
######### #########
FROM image.sourcefind.cn:5000/gpu/admin/base/jupyterlab-pytorch:2.3.1-py3.10-cuda11.8-ubuntu22.04-devel FROM image.sourcefind.cn:5000/gpu/admin/base/jupyterlab-pytorch:2.1.2-py3.10-cuda12.1-ubuntu22.04-devel
ARG IMAGE=nvcomposer ARG IMAGE=nvcomposer
ARG IMAGE_UPPER=NVComposer ARG IMAGE_UPPER=NVComposer
COPY chenyh/$IMAGE/frpc_linux_amd64_* /opt/conda/lib/python3.10/site-packages/gradio/ COPY chenyh/$IMAGE/frpc_linux_amd64_* /opt/conda/lib/python3.10/site-packages/gradio/
......
NVComposer/requirements.txt
View file @ 8c68a142
pytorch_lightning==2.3 pytorch_lightning==2.1.2
deepspeed deepspeed==0.15.0
taming-transformers taming-transformers==0.0.1
scipy diffusers==0.30.1
einops transformers
kornia torchvision==0.16.2
open_clip_torch scipy==1.14.1
openai-clip einops==0.8.0
xformers kornia==0.7.3
timm open_clip_torch==2.20.0
av openai-clip==1.0.1
gradio numpy==1.26.3
\ No newline at end of file xformers==0.0.23.post1
timm==1.0.9
av==12.3.0
gradio==5.8.0
\ No newline at end of file
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