python - 有没有人看到带有 cuml RAPIDS 的错误 global_output_type ？

Question

我是 RAPIDS 的新手，之前只是尝试使用 sklearn 复制逻辑回归模型，但现在使用 cuml，但 LogisticRegression() 调用不起作用。

from cuml.linear_model import LogisticRegression, LinearRegression
ols_cuml=LogisticRegression(penalty='l2')

ols_cuml.fit(X,y)

我收到以下错误：

LinearRegression() 加载语句也得到相同的错误。考虑到 cudf(Pandas for CUDA) 加载良好，并且可以在 gpu 数据帧上运行，不知道为什么会发生这种情况。

score 0 · Accepted Answer

如果构建或 python 设置遇到错误，通常会发生上述错误。

我正在使用 cuml-0.15（每晚发布）并且没有看到与您上面提到的错误相同的错误。尝试从此处提供的命令安装 cuml：https ://rapids.ai/start.html

我使用的是：conda install -c rapidsai-nightly -c nvidia -c conda-forge \ -c defaults cuml=0.15 python=3.7 cudatoolkit=10.2

1 回答 1