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" # --login option used to source bashrc (thus activating conda env) at every RUN statement 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