diff --git a/README.md b/README.md
index 2e01e91..c38576a 100644
--- a/README.md
+++ b/README.md
@@ -22,10 +22,18 @@ DataJoint Elements collectively standardize and automate data collection and ana
   git clone https://github.com/<enter_github_username>/element-moseq.git
   ```
 
++ Install [Conda](https://docs.conda.io/en/latest/miniconda.html).
+
++ Create and activate a Conda environment:
+  ```bash
+   conda env create -f conda_env.yml
+   conda activate element_moseq
+   ```
+
 + Install with `pip`:
 
   ```bash
-  pip install -e .[keypoint-moseq]
+  pip install -e .[kpms_cpu]  # MacOS: pip install -e ".[kpms_cpu]"
   ```
 
 + [Interactive tutorial on GitHub Codespaces](https://github.com/datajoint/element-moseq#interactive-tutorial)
diff --git a/pyproject.toml b/pyproject.toml
index b39d244..e2a16bd 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -34,7 +34,9 @@ elements = [
     "element-interface @ git+https://github.com/datajoint/element-interface.git",
     "element-animal @ git+https://github.com/datajoint/element-animal.git",
 ]
-kpms = ["jax<0.7.0", "keypoint-moseq @ git+https://github.com/dattalab/keypoint-moseq.git"]
+kpms_cpu = ["jax<0.7.0", "keypoint-moseq @ git+https://github.com/dattalab/keypoint-moseq.git"]
+kpms_gpu = ["jax<0.7.0", "keypoint-moseq[cuda] @ git+https://github.com/dattalab/keypoint-moseq.git"]
+
 tests = [
     "pytest",
     "pytest-cov",