Skip to content

Commit

Permalink
Update Dockerfile to activate picssmart environment***
Browse files Browse the repository at this point in the history
  • Loading branch information
Akila-I committed Feb 5, 2024
1 parent 2fa9d01 commit 5b50315
Showing 1 changed file with 16 additions and 6 deletions.
22 changes: 16 additions & 6 deletions Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -4,12 +4,22 @@ FROM condaforge/mambaforge:23.3.1-0
WORKDIR /app

# Install dependencies
RUN mamba create -n picssmart transformers python=3.10 pytorch fairscale dask-mongo torchaudio pytorch-cuda=11.7 iopath cudatoolkit=11.7 -c pytorch -c nvidia -c iopath -c conda-forge -y \
&& conda init bash \
&& source ~/.bashrc \
&& conda activate picssmart \
&& mamba install rapids=23.02 -c rapidsai -c conda-forge -c nvidia -y \
&& pip install "fastapi[all]" pillow pillow-heif einops pycocoevalcap cryptography==38.0.4 motor pymongo pyyaml networkx omegaconf timm decord opencv-python webdataset jupyterlab torchvision tensorflow gdown spacy insightface onnxruntime uvicorn flwr qdrant-client
RUN mamba create -n picssmart transformers python=3.10 pytorch fairscale dask-mongo torchaudio pytorch-cuda=11.7 iopath cudatoolkit=11.7 -c pytorch -c nvidia -c iopath -c conda-forge -y
SHELL ["/bin/bash", "-c"]
RUN conda init bash && source ~/.bashrc
RUN echo "conda activate picssmart" >> ~/.bashrc
ENV PATH /opt/conda/envs/picssmart/bin:$PATH
# RUN conda activate picssmart
RUN mamba install rapids=23.02 -c rapidsai -c conda-forge -c nvidia -y
RUN pip install "fastapi[all]" pillow pillow-heif einops pycocoevalcap cryptography==38.0.4 motor pymongo pyyaml networkx omegaconf timm decord opencv-python webdataset jupyterlab torchvision
RUN pip install tensorflow
RUN pip install gdown
RUN pip install spacy
RUN pip install insightface
RUN pip install onnxruntime
RUN pip install uvicorn
RUN pip install flwr
RUN pip install qdrant-client

# Copy the rest of the application code into the container
COPY . .
Expand Down

0 comments on commit 5b50315

Please sign in to comment.