Add cuda related package to environment for gpu computing

OpenSourceEconomics · Feb 29, 2024 · ae6cd89 · ae6cd89
1 parent 9a9b874
commit ae6cd89
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/docs/source/conf.py b/docs/source/conf.py
@@ -49,7 +49,6 @@
     "pytest",
     "scipy",
     "jax",
-    "estimagic",
 ]
 
 # Add any paths that contain templates here, relative to this directory.

diff --git a/environment.yml b/environment.yml
@@ -3,6 +3,7 @@ name: lcm
 channels:
   - conda-forge
   - nodefaults
+  - nvidia
 dependencies:
   - python=3.11
   - pip
@@ -26,7 +27,7 @@ dependencies:
   - pytest-xdist
 
   # Developer dependencies
-  - estimagic
+  - cuda-nvcc
   - jupyterlab
   - matplotlib
   - numba