From a86495cf6b809ea490059c2150b88801c3234fe2 Mon Sep 17 00:00:00 2001
From: Luca Grillotti <luca.grillotti16@imperial.ac.uk>
Date: Wed, 28 Aug 2024 09:08:33 +0000
Subject: [PATCH] added instructions for pip install qdax[cuda12]

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

diff --git a/README.md b/README.md
index 551680eb..dab09614 100644
--- a/README.md
+++ b/README.md
@@ -24,6 +24,12 @@ QDax is available on PyPI and can be installed with:
 ```bash
 pip install qdax
 ```
+
+To install QDax with CUDA 12 support, use:
+```bash
+pip install qdax[cuda12]
+```
+
 Alternatively, the latest commit of QDax can be installed directly from source with:
 ```bash
 pip install git+https://github.com/adaptive-intelligent-robotics/QDax.git@main