Skip to content

Commit

Permalink
restart shell after conda install
Browse files Browse the repository at this point in the history
  • Loading branch information
Akila-I committed Feb 5, 2024
1 parent 2a63ecb commit 356dad8
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@ WORKDIR /app
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 exec $SHELL
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
Expand All @@ -26,4 +27,4 @@ COPY . .
EXPOSE 8000

# Command to run the application
CMD ["uvicorn", "server.__main__:create_app", "--factory", "--reload"]
CMD ["uvicorn", "server.__main__:create_app", "--factory", "--reload"]

0 comments on commit 356dad8

Please sign in to comment.