Update README.md

raucha · Sep 24, 2023 · e0fd92d · e0fd92d
1 parent 22cf461
commit e0fd92d
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -6,8 +6,27 @@
 
 ## Usage
 
+### CPU
 ```json
 "features": {
   "ghcr.io/raucha/devcontainer-features/pytorch:latest": {}
 }
 ```
+
+### GPU
+```
+"features": {
+  "ghcr.io/devcontainers/features/nvidia-cuda:1": {
+    "installCudnn": true,
+    "cudaVersion": "11.8",
+    "cudnnVersion": "8.6.0.163"
+  },
+  "ghcr.io/raucha/devcontainer-features/pytorch:latest": {}
+},
+"runArgs": [
+  "--gpus",
+  "all",
+  "--shm-size",
+  "16gb"
+],
+```