diff --git a/grand_challenge_forge/partials/example-algorithm/example-algorithm{{cookiecutter._}}/Dockerfile b/grand_challenge_forge/partials/example-algorithm/example-algorithm{{cookiecutter._}}/Dockerfile index 7953241..22b46d7 100644 --- a/grand_challenge_forge/partials/example-algorithm/example-algorithm{{cookiecutter._}}/Dockerfile +++ b/grand_challenge_forge/partials/example-algorithm/example-algorithm{{cookiecutter._}}/Dockerfile @@ -1,4 +1,4 @@ -FROM --platform=linux/amd64 pytorch/pytorch +FROM pytorch/pytorch # Use a 'large' base container to show-case how to load pytorch and use the GPU (when enabled) # Ensures that Python output to stdout/stderr is not buffered: prevents missing information when terminating diff --git a/grand_challenge_forge/partials/example-evaluation-method/example-evaluation-method{{cookiecutter._}}/Dockerfile b/grand_challenge_forge/partials/example-evaluation-method/example-evaluation-method{{cookiecutter._}}/Dockerfile index 6bb0424..04bae65 100644 --- a/grand_challenge_forge/partials/example-evaluation-method/example-evaluation-method{{cookiecutter._}}/Dockerfile +++ b/grand_challenge_forge/partials/example-evaluation-method/example-evaluation-method{{cookiecutter._}}/Dockerfile @@ -1,4 +1,4 @@ -FROM --platform=linux/amd64 docker.io/library/python:3.11-slim +FROM docker.io/library/python:3.11-slim # Ensures that Python output to stdout/stderr is not buffered: prevents missing information when terminating ENV PYTHONUNBUFFERED=1