➕ Add cuml and scikit-learn

GPU Machine Learning Algorithms, in Python! This also brings in cudf - a GPU DataFrame Library, cudatoolkit, and other RAPIDS AI libraries. Pinned numba to 0.51.0 and bumped up llvmlite in the poetry.lock file to 0.34.0 to be in line with conda installed version and prevent "ERROR: Cannot uninstall 'llvmlite'. It is a distutils installed project and thus we cannot accurately determine which files belong to it which would lead to only a partial uninstall". This requires datashader 0.11.1 too, all in the fun of dependency management.
weiji14 · Sep 15, 2020 · cbd1541 · cbd1541
1 parent 246aae3
commit cbd1541
Show file tree

Hide file tree

Showing 3 changed files with 119 additions and 65 deletions.
diff --git a/environment.yml b/environment.yml
@@ -1,11 +1,14 @@
 name: deepicedrain
 channels:
   - conda-forge
+  - rapidsai-nightly
 dependencies:
+  - rapidsai-nightly::cuml=0.15.0a200817[md5=8e6ede59b1bc5742e21e1e5327173a8b]
   - conda-forge::geos=3.8.1[md5=7282fbe0c036bacad959c03c05a7eb9a]
   - conda-forge/label/dev::gmt=6.2.0.dev0+a1a189c[md5=915cb603613887002b58e09859aaac82]
   - conda-forge::gxx_linux-64=7.5.0[md5=c6afca6a2bfe1ea0ff5566d300460501]
   - conda-forge::nodejs=13.13.0[md5=cf25f8cee7aad7f417980872d34d8fd6]
+  - numba::numba=0.51.0[md5=1151f4a9312696ac4694d82089c92a11]
   - conda-forge::numcodecs=0.6.4[md5=4873a5169bb684b4e120e7bcb287488e]
   - conda-forge::parallel=20200522[md5=8fa06db0afc1e7b1f136277557a4b65e]
   - conda-forge::pip=20.1.1[md5=5e42afe5672d5baf3d9d2260239aa923]