File size: 2,185 Bytes
8c92a11 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 |
# Copyright (c) 2023 Amphion.
#
# This source code is licensed under the MIT license found in the
# LICENSE file in the root directory of this source tree.
# Other version: https://hub.docker.com/r/nvidia/cuda/tags
FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu18.04
ARG DEBIAN_FRONTEND=noninteractive
ARG PYTORCH='2.0.0'
ARG CUDA='cu118'
ARG SHELL='/bin/bash'
ARG MINICONDA='Miniconda3-py39_23.3.1-0-Linux-x86_64.sh'
ENV LANG=en_US.UTF-8 PYTHONIOENCODING=utf-8 PYTHONDONTWRITEBYTECODE=1 CUDA_HOME=/usr/local/cuda CONDA_HOME=/opt/conda SHELL=${SHELL}
ENV PATH=$CONDA_HOME/bin:$CUDA_HOME/bin:$PATH \
LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH \
LIBRARY_PATH=$CUDA_HOME/lib64:$LIBRARY_PATH \
CONDA_PREFIX=$CONDA_HOME \
NCCL_HOME=$CUDA_HOME
# Install ubuntu packages
RUN sed -i 's/archive.ubuntu.com/mirrors.cloud.tencent.com/g' /etc/apt/sources.list \
&& sed -i 's/security.ubuntu.com/mirrors.cloud.tencent.com/g' /etc/apt/sources.list \
&& rm /etc/apt/sources.list.d/cuda.list \
&& apt-get update \
&& apt-get -y install \
python3-pip ffmpeg git less wget libsm6 libxext6 libxrender-dev \
build-essential cmake pkg-config libx11-dev libatlas-base-dev \
libgtk-3-dev libboost-python-dev vim libgl1-mesa-glx \
libaio-dev software-properties-common tmux \
espeak-ng
# Install miniconda with python 3.9
USER root
# COPY Miniconda3-py39_23.3.1-0-Linux-x86_64.sh /root/anaconda.sh
RUN wget -t 0 -c -O /tmp/anaconda.sh https://repo.anaconda.com/miniconda/${MINICONDA} \
&& mv /tmp/anaconda.sh /root/anaconda.sh \
&& ${SHELL} /root/anaconda.sh -b -p $CONDA_HOME \
&& rm /root/anaconda.sh
RUN conda create -y --name amphion python=3.9.15
WORKDIR /app
COPY env.sh env.sh
RUN chmod +x ./env.sh
RUN ["conda", "run", "-n", "amphion", "-vvv", "--no-capture-output", "./env.sh"]
RUN conda init \
&& echo "\nconda activate amphion\n" >> ~/.bashrc
CMD ["/bin/bash"]
# *** Build ***
# docker build -t realamphion/amphion .
# *** Run ***
# cd Amphion
# docker run --runtime=nvidia --gpus all -it -v .:/app -v /mnt:/mnt_host realamphion/amphion
# *** Push and release ***
# docker login
# docker push realamphion/amphion
|