|
|
|
FROM nvidia/cuda:11.7.1-cudnn8-devel-ubuntu22.04 |
|
|
|
|
|
RUN apt-get update && apt-get install -y wget git |
|
|
|
|
|
RUN wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh && \ |
|
bash Miniconda3-latest-Linux-x86_64.sh -b -p /opt/conda && \ |
|
rm Miniconda3-latest-Linux-x86_64.sh |
|
|
|
|
|
ENV PATH /opt/conda/bin:$PATH |
|
|
|
|
|
RUN git clone https://github.com/LLaVA-VL/LLaVA-Interactive-Demo.git |
|
|
|
|
|
RUN conda create -n llava_int -c conda-forge -c pytorch python=3.10.8 pytorch=2.0.1 -y |
|
|
|
|
|
SHELL ["conda", "run", "-n", "llava_int", "/bin/bash", "-c"] |
|
|
|
|
|
WORKDIR /LLaVA-Interactive-Demo |
|
|
|
|
|
RUN pip install -r requirements.txt |
|
|
|
|
|
RUN source setup.sh |
|
|
|
|
|
|
|
CMD ["./run_demo.sh"] |
|
|