Anaconda Xgboost unable to find GPU

Viewed 437

It seems that Anaconda is unable recognise my GPU, GPU is RTX2070 (Driver version 510.47.03), system Ubuntu 20.04, cudatoolkit 11.3.1, cudnn 8.2.1, XGboost 1.5.2 via pip install.

When I run XGboost with GPU enable it shows:

XGBoostError: [01:24:12] ../src/gbm/gbtree.cc:531: Check failed: common::AllVisibleGPUs() >= 1 (0 vs. 1) : No visible GPU is found for XGBoost.

I also used GPUtil to check the visible GPU, it is showing 0 GPU.

Below is the output from nvidia-smi

enter image description here

Can anyone help? how come the GPU in the system is no visible to any package in anaconda? (Pytorch, XGBoost etc.)

0 Answers
Related