Merge branch 'installation_refactor'

dattalab · Oct 18, 2024 · 0800b57 · 0800b57
2 parents 2a1cf00 + e91ef9e
commit 0800b57
Show file tree

Hide file tree

Showing 5 changed files with 718 additions and 679 deletions.
diff --git a/conda_envs/environment.linux_gpu.yml b/conda_envs/environment.linux_gpu.yml
@@ -14,5 +14,6 @@ dependencies:
   - pip
   - pip:
     - "keypoint-moseq"
+    - "jax-moseq[cuda11]"
     - jupyterlab
     - etils==1.5.2
diff --git a/conda_envs/environment.win64_gpu.yml b/conda_envs/environment.win64_gpu.yml
@@ -16,4 +16,5 @@ dependencies:
     - jax==0.3.22
     - "https://whls.blob.core.windows.net/unstable/cuda111/jaxlib-0.3.22+cuda11.cudnn82-cp39-cp39-win_amd64.whl"
     - "keypoint-moseq"
+    - "jax-moseq[cuda11]"
     - jupyterlab