Skip to content

Instantly share code, notes, and snippets.

@elviejokike
Last active August 14, 2020 08:36
Show Gist options
  • Save elviejokike/4b9ca2e7e1172c9a121843adb5fd25f9 to your computer and use it in GitHub Desktop.
Save elviejokike/4b9ca2e7e1172c9a121843adb5fd25f9 to your computer and use it in GitHub Desktop.
Dockerfile-4Sagemaker
FROM 763104351884.dkr.ecr.eu-west-1.amazonaws.com/pytorch-training:1.5.1-gpu-py36-cu101-ubuntu16.04
RUN mkdir -p /opt/service/boot
COPY entrypoint.sh /opt/service/boot/
RUN chmod +x /opt/service/boot/entrypoint.sh
WORKDIR /opt/service/boot
ENTRYPOINT ["/opt/service/boot/entrypoint.sh"]
#!/bin/bash
set -e
source /opt/ml/input/data/service-boot/job.sh
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment