python - 如何安装 cuML 模块？

Question

我在我的 jupyter 笔记本中启动了这个命令

pip install cuml

我有这个错误：

错误：找不到满足 cuml 要求的版本（来自版本：无） 错误：没有找到 cuml 的匹配分布

请问我该如何解决（我在windows上）？

Answer 1

抱歉，cuml 目前不支持 windows 操作系统。但是您可以在云端免费试用。

Kaggle 内核：https ://www.kaggle.com/cdeotte/rapids-gpu-knn-mnist-0-97

Colab：https ://colab.research.google.com/drive/1rY7Ln6rEE1pOlfSHCYOVaqt8OvDO35J0#forceEdit=true&offline=true&sandboxMode=true

让我知道这是否适合你。感谢您对 cuml 感兴趣！

Answer 2

您还可以使用 docker 命令拉取已安装 RAPIDS 库套件的映像。

例子：

docker pull rapidsai/rapidsai:cuda10.0-runtime-ubuntu16.04

docker run --runtime nvidia --rm -it -p 8888:8888 -p 8787:8787 -p 8786:8786 \
    rapidsai/rapidsai:cuda10.0-runtime-ubuntu16.04

更多详情：https ://rapids.ai/start.html