ubuntu 18 安装xgboost GPU版本

先把之前安装的xgboost都卸载干净：

pip uninstall xgboost

确保电脑里安装了cuda。

然后按照下面的步骤来安装GPU版本的xgboost：

git clone --recursive https://github.com/dmlc/xgboost
cd xgboost
mkdir build
cd build
cmake .. -DUSE_CUDA=ON
make -j
sh build.sh
cd python-package
python setup.py install

测试GPU
在tests\benchmark目录下
分别执行

python benchmark.py --tree_method gpu_hist
python benchmark.py --tree_method hist

可以看到，第一个gpu版本的比第二个非GPU版本的快。在实际的代码中，只要在添加下面的参数就行了

param['tree_method'] = 'gpu_hist'

https://www.cnblogs.com/huadongw/p/6161145.html

https://blog.csdn.net/u011587516/article/details/78995186

ubuntu 18 安装xgboost GPU版本相关推荐

WINDOWS 安装XGBoost GPU版本最新简易方法
目录一.系统配置二.问题背景三.执行步骤 1. 安装cuda a. 检查是否安装了CUDA b. 从dos中查看可以支持的cuda版本 c 下载对应版本的cuda d 根据引导安装cuda 2. ...
Ubuntu 20.04 安装Tensorflow GPU版本（GTX-1060显卡）
Ubuntu 20.04 安装Tensorflow GPU版本 (GTX-1060显卡) 参考: https://cyfeng.science/2020/05/02/ubuntu-install-nv ...
windows10下安装tensorflow(gpu版本)
windows10安装tensorflow的gpu版本(pip3安装方式) 前言: TensorFlow 有cpu和 gpu两个版本:gpu版本需要英伟达CUDA 和 cuDNN 的支持,cpu版本不 ...
2020最新ubuntu 18 安装emqttd(emqx)
2020最新linux ubuntu 18 安装emqttd(emqx) 最近需要在linux ubuntu中安装MQTT服务,网上找了很多安装步骤,发现都不好用,发现EMQ近期有过一次更新,所以以前 ...
ubuntu查看安装软件的版本
ubuntu查看安装软件的版本 dpkg -l libturbojpeg* libjpeg*
Ubuntu 18 安装截图工具 flameshot
一. 安装flameshot Ubuntu 18 安装截图工具非常简单,只需一条指令: sudo apt-get install flameshot 然后终端执行 flameshot gui 拖动鼠 ...
rbx1 ubuntu 18安装
rbx1 ubuntu 18安装 sudo apt-get install ros-melodic-arbotix ros-melodic-openni-camera \ ros-melodic-ro ...
pp-tracking ubuntu 18.04使用GPU推理出现Segmentation fault
目录 1. 报错 2. 报错原因 3. conda环境指定cuda 10.2和cudnn 7.6.5 3.1. 安装cuda 10.2 3.2. 安装cudnn 7.6.5 3.3. conda环境指 ...
UCloud Ubuntu 18.04 配置GPU环境踩坑指南，包括驱动，CUDA，Docker，Nvidia-Docker等
设备:NVIDIA Corporation GV100GL [Tesla V100 PCIe 16GB] (rev a1) # Ucloud Ubuntu 18.04 GPU 环境 Driver,Cu ...

ubuntu 18 安装xgboost GPU版本

ubuntu 18 安装xgboost GPU版本相关推荐

最新文章

热门文章