aihub-dag/etc/config/metadata/runtime-images/pytorch-devel.json

{
  "display_name": "Pytorch 1.4 with CUDA-devel",
  "metadata": {
    "description": "PyTorch 1.4 (with GPU support)",
    "image_name": "pytorch/pytorch:1.4-cuda10.1-cudnn7-devel",
    "tags": ["gpu", "pytorch"]
  },
  "schema_name": "runtime-image"
}