Skip to content

Commit b06abd2

Browse files
committed
add libtcmalloc-minimal4 in Dockerfile
1 parent 1824b23 commit b06abd2

File tree

1 file changed

+1
-1
lines changed
  • inference/generativeai/llm-workshop/lab12-hosting-controlnet-models-on-sagemaker

1 file changed

+1
-1
lines changed

inference/generativeai/llm-workshop/lab12-hosting-controlnet-models-on-sagemaker/Dockerfile.inference

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
ARG REGION
22
FROM 763104351884.dkr.ecr.$REGION.amazonaws.com/pytorch-inference:2.0.1-gpu-py310-cu118-ubuntu20.04-sagemaker
33

4-
RUN apt update && apt install wget git libgl1-mesa-glx libcairo2-dev -y
4+
RUN apt update && apt install wget git libgl1-mesa-glx libcairo2-dev libtcmalloc-minimal4 -y
55
RUN rm -rf /opt/conda/lib/python3.10/site-packages/opencv_python-4.6.0.dist-info/
66
RUN pip install opencv-python
77
RUN pip install xformers==0.0.22.post7 --extra-index-url https://download.pytorch.org/whl/cu118

0 commit comments

Comments
 (0)