added instructions for pip install qdax[cuda12]

adaptive-intelligent-robotics · Aug 28, 2024 · a86495c · a86495c
1 parent bb46a5e
commit a86495c
Showing 1 changed file with 6 additions and 0 deletions.
diff --git 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