From 6af36cffec72cdb9ecefd08564aba0759a4ef4c3 Mon Sep 17 00:00:00 2001
From: Jacob Marshall <me@jhqcat.com>
Date: Thu, 15 Feb 2024 23:39:09 +0000
Subject: [PATCH] require cuda

---
 .devcontainer/devcontainer.json | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/.devcontainer/devcontainer.json b/.devcontainer/devcontainer.json
index 8a7d6b114..27d97ae29 100644
--- a/.devcontainer/devcontainer.json
+++ b/.devcontainer/devcontainer.json
@@ -3,6 +3,11 @@
   "hostRequirements": {
     "cpus": 4
   },
+  "features": {
+    "ghcr.io/devcontainers/features/nvidia-cuda:1": {
+      "installCudnn": true
+    }
+  },
   "waitFor": "onCreateCommand",
   "updateContentCommand": "python3 -m pip install -r requirements.txt",
   "postCreateCommand": "",