|
FROM nvidia/cuda:12.2.0-base-ubuntu22.04 |
|
|
|
COPY . /app |
|
|
|
RUN apt-get update && \ |
|
apt-get install -y --allow-unauthenticated --no-install-recommends \ |
|
wget \ |
|
git \ |
|
&& apt-get autoremove -y \ |
|
&& apt-get clean -y \ |
|
&& rm -rf /var/lib/apt/lists/* |
|
|
|
ENV HOME "/root" |
|
ENV CONDA_DIR "${HOME}/miniconda" |
|
ENV PATH="$CONDA_DIR/bin":$PATH |
|
ENV CONDA_AUTO_UPDATE_CONDA=false |
|
ENV PIP_DOWNLOAD_CACHE="$HOME/.pip/cache" |
|
ENV TORTOISE_MODELS_DIR |
|
|
|
RUN wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O /tmp/miniconda3.sh \ |
|
&& bash /tmp/miniconda3.sh -b -p "${CONDA_DIR}" -f -u \ |
|
&& "${CONDA_DIR}/bin/conda" init bash \ |
|
&& rm -f /tmp/miniconda3.sh \ |
|
&& echo ". '${CONDA_DIR}/etc/profile.d/conda.sh'" >> "${HOME}/.profile" |
|
|
|
|
|
SHELL ["/bin/bash", "--login", "-c"] |
|
|
|
RUN conda create --name tortoise python=3.9 numba inflect \ |
|
&& conda activate tortoise \ |
|
&& conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia \ |
|
&& conda install transformers=4.29.2 \ |
|
&& cd /app \ |
|
&& python setup.py install |
|
|
