docker.json

{
  "variables": {
    "anaconda_dir": "/opt/anaconda",
    "anaconda_env": "default",
    "cache": null,
    "tag": null
  },
  "builders": [
    {
      "type": "docker",
      "image": "nvidia/cuda:9.2-cudnn7-devel-ubuntu16.04",
      "commit": true,
      "volumes": {
        "{{pwd}}": "/env",
        "{{user `cache`}}/install": "/install",
        "{{user `cache`}}/pkgs": "{{user `anaconda_dir`}}/pkgs"
      },
      "changes": [
        "ENV ANACONDA_DIR {{user `anaconda_dir`}}",
        "ENV ANACONDA_ENV {{user `anaconda_env`}}",
        "ENTRYPOINT [\"/entrypoint\"]",
        "WORKDIR /src",
        "ONBUILD COPY environment.yml ./",
        "ONBUILD RUN {{user `anaconda_dir`}}/bin/conda env update --name={{user `anaconda_env`}}",
        "ONBUILD COPY . ./"
      ]
    }
  ],
  "provisioners": [
    {
      "type": "shell",
      "inline_shebang": "/bin/sh -eux",
      "inline": [
        "apt-get update -qq",
        "apt-get install -y bzip2 curl g++",
        "/install/anaconda.sh -b -u -p {{user `anaconda_dir`}}",
        "{{user `anaconda_dir`}}/bin/conda env create --name={{user `anaconda_env`}} --file=/env/environment.yml",
        "{{user `anaconda_dir`}}/bin/conda update -y -n base conda"
      ]
    },
    {
      "type": "file",
      "source": "entrypoint",
      "destination": "/entrypoint"
    }
  ],
  "post-processors": [
    [
      {
        "type": "docker-tag",
        "repository": "{{user `tag`}}",
        "tag": "latest"
      }
    ]
  ]
}