{
  "build": [
    {
      "torch_command": "pip install torch==2.3.0 torchvision==0.18.0 torchaudio==2.3.0 --index-url https://download.pytorch.org/whl/cu121",
      "cuda_image": "hpcaitech/cuda-conda:12.1"
    },
    {
      "torch_command": "pip install torch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1 --index-url https://download.pytorch.org/whl/cu124",
      "cuda_image": "hpcaitech/cuda-conda:12.4"
    }
  ]
}