MuseV/MMCM/Dockerfile

# FROM mirrors.tencent.com/todacc/venus-std-base-cuda11.8:0.1.0
FROM mirrors.tencent.com/todacc/venus-std-ext-cuda11.8-pytorch2.0-tf2.12-py3.10:0.7.0

#MAINTAINER 维护者信息
LABEL MAINTAINER="anchorxia"
LABEL Email="[email protected]"
LABEL Description="gpu development image, from mirrors.tencent.com/todacc/venus-std-ext-cuda11.8-pytorch2.0-tf2.12-py3.10:0.7.0"

USER root
# 安装必须软件
# RUN GENERIC_REPO_URL="http://mirrors.tencent.com/repository/generic/venus_repo/image_res" \
# && cd /data/ \
# && wget -q $GENERIC_REPO_URL/gcc/gcc-11.2.0.zip \
# && unzip -q gcc-11.2.0.zip \
# && cd gcc-releases-gcc-11.2.0 \
# && ./contrib/download_prerequisites \
# && ./configure --enable-bootstrap --enable-languages=c,c++ --enable-threads=posix --enable-checking=release --enable-multilib --with-system-zlib \
# && make --silent -j10 \
# && make --silent install \
# && gcc -v \
# && rm -rf /data/gcc-releases-gcc-11.2.0 /data/gcc-11.2.0.zip 

# RUN yum update -y \
# && yum install -y epel-release \
# && yum install -y ffmpeg \
# && yum install -y Xvfb \
# && yum install -y centos-release-scl devtoolset-11
RUN yum install -y wget zsh git curl tmux cmake htop iotop git-lfs zip \
 && yum install -y autojump autojump-zsh portaudio portaudio-devel \
 && yum clean all

USER mqq
RUN source ~/.bashrc \
 && GENERIC_REPO_URL="http://mirrors.tencent.com/repository/generic/venus_repo/image_res" \
 && conda deactivate \
 # && conda remove -y -n env-2.7.18 --all \
 # && conda remove -y -n env-3.6.8 --all \
 # && conda remove -y -n env-3.7.7 --all \
 # && conda remove -y -n env-3.8.8 --all \
 # && conda remove -y -n env-3.9.2 --all \
 # && conda remove -y -n env-novelai --all \
 && conda create -n projectv python=3.10.6 -y \
 && conda activate projectv \
 && pip install venus-sdk -q -i https://mirrors.tencent.com/repository/pypi/tencent_pypi/simple \
 --extra-index-url https://mirrors.tencent.com/pypi/simple/ \
 && pip install tensorflow==2.12.0 tensorboard==2.12.0 \
 && pip install torch==2.0.1+cu118 torchvision==0.15.2+cu118 -f https://mirror.sjtu.edu.cn/pytorch-wheels/torch_stable.html -i https://mirrors.bfsu.edu.cn/pypi/web/simple -U \
 # 安装xformers，支持不同型号gpu
 && pip install ninja==1.11.1 \
 # && git clone https://github.com/facebookresearch/xformers.git \
 # && cd xformers \
 # && git checkout v0.0.17rc482 \
 # && git submodule update --init --recursive \
 # && pip install numpy==1.23.4 pyre-extensions==0.0.23 \
 # && FORCE_CUDA="1" MAX_JOBS=1 TORCH_CUDA_ARCH_LIST="6.1;7.0;7.5;8.0;8.6" pip install -e . \
 # && cd .. \
 # 安装一堆包
 && pip install --no-cache-dir transformers bitsandbytes decord accelerate xformers omegaconf einops imageio==2.31.1 \
 && pip install --no-cache-dir pandas h5py matplotlib modelcards pynvml black pytest moviepy torch-tb-profiler scikit-learn librosa ffmpeg easydict webp controlnet_aux mediapipe \
 && pip install --no-cache-dir Cython easydict gdown infomap insightface ipython librosa onnx onnxruntime onnxsim opencv_python Pillow protobuf pytube PyYAML \
 && pip install --no-cache-dir requests scipy six tqdm gradio albumentations opencv-contrib-python imageio-ffmpeg pytorch-lightning test-tube \
 && pip install --no-cache-dir timm addict yapf prettytable safetensors basicsr fvcore pycocotools wandb gunicorn \
 && pip install --no-cache-dir streamlit webdataset kornia open_clip_torch streamlit-drawable-canvas torchmetrics \
 # 安装暗水印
 && pip install --no-cache-dir invisible-watermark==0.1.5 gdown==4.5.3 ftfy==6.1.1 modelcards==0.1.6 \
 # 安装openmm相关包
 && pip install--no-cache-dir -U openmim \
 && mim install mmengine \
 && mim install "mmcv>=2.0.1" \
 && mim install "mmdet>=3.1.0" \
 && mim install "mmpose>=1.1.0" \
 # jupyters
 && pip install ipywidgets==8.0.3 \
 && python -m ipykernel install --user --name projectv --display-name "python(projectv)" \
 && pip install --no-cache-dir matplotlib==3.6.2 redis==4.5.1 pydantic[dotenv]==1.10.2 loguru==0.6.0 IProgress==0.4 \
 && pip install --no-cache-dir cos-python-sdk-v5==1.9.22 coscmd==1.8.6.30 \
 # 必须放在最后pip，避免和jupyter的不兼容
 && pip install --no-cache-dir markupsafe==2.0.1 \
 && wget -P /tmp $GENERIC_REPO_URL/cpu/clean-layer.sh \
 && sh /tmp/clean-layer.sh

ENV LD_LIBRARY_PATH=/usr/local/lib64:$LD_LIBRARY_PATH
USER root