[SOLVED] Anaconda Xgboost unable to find GPU

Issue

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

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

Solution

Appeared to be a driver issue, tried the same code in another computer with two RTX 3090 and got no issue

Answered By - tgtm005

Answer Checked By - Marilyn (WPSolving Volunteer)