From e0fd92d055f3710ee9813d4d38bca7e9fea865f7 Mon Sep 17 00:00:00 2001
From: raucha <raucha@users.noreply.github.com>
Date: Sun, 24 Sep 2023 12:21:28 +0900
Subject: [PATCH] Update README.md

---
 README.md | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/README.md b/README.md
index 49f72e0..44132a2 100644
--- 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"
+],
+```