MuseV/Dockerfile
xzqjack 504d705db4
update gradio (#55)
* fix limitation of gpu vram

* aad time limitation

* reshape logci

* <feature> add example and limitation of input

* update gradio scripts and requirement

---------

Co-authored-by: trumpzhan <trumpzhan@tencent.com>
2024-04-10 16:08:45 +08:00

19 lines
527 B
Docker
Executable File

FROM anchorxia/musev:1.0.0
#MAINTAINER 维护者信息
LABEL MAINTAINER="anchorxia"
LABEL Email="anchorxia@tencent.com"
LABEL Description="musev gpu runtime image, base docker is pytorch/pytorch:2.0.1-cuda11.7-cudnn8-devel"
ARG DEBIAN_FRONTEND=noninteractive
USER root
SHELL ["/bin/bash", "--login", "-c"]
RUN . /opt/conda/etc/profile.d/conda.sh \
&& echo "source activate musev" >> ~/.bashrc \
&& conda activate musev \
&& conda env list \
&& pip --no-cache-dir install cuid gradio==4.12 spaces
USER root