spam-ml-mx-v2

commit 8609c7af1dbd4d7b640ce220677a38fbb7ef1dc2
parent cae439fbb240be414d3bde9fcf780589177a1fad
Author: Marcel <MTRNord@users.noreply.github.com>
Date:   Fri, 10 Mar 2023 14:29:40 +0000

Update codespace config and add requirements.txt

Diffstat:
M
.devcontainer/devcontainer.json
 | 
9
++++++---
A
requirements.txt
 | 
6
++++++

2 files changed, 12 insertions(+), 3 deletions(-)
diff --git a/.devcontainer/devcontainer.json b/.devcontainer/devcontainer.json
@@ -11,7 +11,10 @@
     "features": {
         "ghcr.io/devcontainers/features/nvidia-cuda:1": {
             "installCudnn": true,
-            "cudaVersion": "11.7"
-        }
-    }
+            "cudaVersion": "11.7",
+            "cudnnVersion": "8.5.0.96"
+        },
+        "ghcr.io/devcontainers/features/python:1": {}
+    },
+    "postCreateCommand": "pip3 install --user -r requirements.txt"
 }
 \ No newline at end of file
diff --git a/requirements.txt b/requirements.txt
@@ -0,0 +1,5 @@
+torch
+torchaudio
+torchvision
+fastai
+numpy
+\ No newline at end of file