Search code examples
pythonubuntupytorchanacondagpu

Anaconda Xgboost unable to find GPU


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.)


Solution

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