RuntimeError:使用cuml时遇到的CUDA错误

Question

当我使用rapids时，我总是会遇到错误：

现在，我跑：

from cuml.datasets.regression import make_regression
data, values = make_regression(n_samples=200000, n_features=50,
                               n_informative=7, bias=-4.2,
                               noise=0.3, random_state=10)

错误信息是：

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/opt/conda/envs/rapids/lib/python3.9/contextlib.py", line 79, in inner
    return func(*args, **kwds)
  File "/opt/conda/envs/rapids/lib/python3.9/site-packages/cuml/internals/api_decorators.py", line 473, in inner
    ret_val = func(*args, **kwargs)
  File "regression.pyx", line 186, in cuml.datasets.regression.make_regression
  File "handle.pyx", line 63, in pylibraft.common.handle.Handle.__cinit__
RuntimeError: CUDA error encountered at: file=/opt/conda/envs/rapids/include/raft/core/handle.hpp line=301:

我使用WLS2 Ubuntu20.04和docker

MAND
docker pull nvcr.io/nvidia/rapidsai/rapidsai-core:22.10-cuda11.5-runtime-ubuntu20.04-py3.9
docker run --gpus all --rm -it \
    --shm-size=1g --ulimit memlock=-1 \
    -p 8888:8888 -p 8787:8787 -p 8786:8786 \
    nvcr.io/nvidia/rapidsai/rapidsai-core:22.10-cuda11.5-runtime-ubuntu20.04-py3.9

我的GPU是nvidia GeForece 1080。

我不知道有什么问题。

Answer 1

在WSL 2上，Rapid22.10需要一个具有计算能力7.0或更高的GPU。在https://rapids.ai/wsl2.html的安装指南中记录了这个要求。

GeForce 1080具有计算功能6.1，目前不支持。有关计算能力的其他参考资料，请参阅https://developer.nvidia.com/cuda-gpus。RAPIDS在WSL 2上支持RTX 2000系列或更高版本。或者，您可以使用GTX 1080与本机安装Ubuntu，而不是通过WSL 2安装Windows。

谢谢你使用急流！