You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Thanks for the issue @Nithish-Chowdary, the ussue is that CUDA 9.2 is not supported, the packages that are being installed require CUDA 11.2 as a minimum, with the cu12 versions being usable in any CUDA 12.0 system.
Unfortunately, the version provided by nvidia-smi just corresponds to the driver's CUDA version (i.e. the CUDA that came bundled with the driver), and doesn't necessarily correspond with the CTK of an environment. For example with conda one could have a driver for CUDA 12.x and have CUDA 11.x in the conda env. Since you are using pip to install, and cu11 then you'd need a CUDA 11, you can check that probably by typing nvcc --version in the command line
Describe the bug
I'm trying to build a Random Forest Regressor using cuml.dask.ensemble.RandomForestRegressor.
collab V100 GPUs are used.
Steps/Code to reproduce bug
from cuml.dask.ensemble import RandomForestRegressor as cuRF
from dask_ml.model_selection import train_test_split
train, test = train_test_split(Data, test_size=0.2, shuffle=True, random_state=42)
train_labels = train[single_label].map_partitions(cudf.from_pandas)
train_features = train[target_features].map_partitions(cudf.from_pandas)
test_labels = test[single_label].map_partitions(cudf.from_pandas)
test_features = test[target_features].map_partitions(cudf.from_pandas)
model = cuRF(n_estimators=100, random_state=42, verbose = True)
model.fit(train_features, train_labels)
Expected behavior
I want the model to fit
Environment details (please complete the following information):
!pip install cudf-cu11 dask-cudf-cu11 cuml-cu11 cugraph-cu11 --extra-index-url=https://pypi.ngc.nvidia.com
!rm -rf /usr/local/lib/python3.8/dist-packages/cupy*
!pip install cupy-cuda11x
The text was updated successfully, but these errors were encountered: