Skip to content

Instantly share code, notes, and snippets.

@tsh-code

tsh-code/Dockerfile

Created May 6, 2024 09:32
Show Gist options
  • Save tsh-code/a739c16fe3312d46400d9dc96a8eaf66 to your computer and use it in GitHub Desktop.
Save tsh-code/a739c16fe3312d46400d9dc96a8eaf66 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
Dockerfile
FROM amazon/aws-lambda-python:3.9
COPY ./requirements.txt .
RUN yum -y install gcc-c++
RUN pip install --no-cache-dir torch --extra-index-url https://download.pytorch.org/whl/cpu
RUN pip install --no-cache-dir -r requirements.txt
ENV HOME /tmp
COPY handler.py ${LAMBDA_TASK_ROOT}
CMD ["handler.main"]
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment