admin管理员组

文章数量:1632336

目录

      • 一、系统配置
      • 二、问题背景
      • 三、执行步骤
        • 1. 安装cuda
            • a. 检查是否安装了CUDA
            • b. 从dos中查看可以支持的cuda版本
            • c 下载对应版本的cuda
            • d 根据引导安装cuda
        • 2. 安装XGBoost
            • a 下载对应的whl文件
            • b 使用whl文件进行xgb的安装
        • 3. 运行官方测试用例

一、系统配置

系统:Windows 10 64x

二、问题背景

  1. 在网上打比赛的时候通常会用到XGBoost算法,但是在数据量大或者特征多的情况下,XGBOOST的CPU版本通常会吃不消,这个时候就需要用XGBoost的GPU版本来加速一下运算速度(但是经过测试,有有时候使用gpu加速之后还不如默认的快,希望大佬可以分享自己的看法!)。
  2. 网上关于安装XGBoost-GPU加速的博文大多过于复杂,需要使用Cmake进行编译。而实际上,我们使用pip进行安装的就是可以支持GPU加速的,官网(官网指导界面)解释如下:

三、执行步骤

1. 安装cuda
a. 检查是否安装了CUDA

在dos中输入 nvcc --version 检查是否安装了cuda,如果有输出那就是安装了,则转到 2点击跳转中进行安装xgb即可,反之如果输出没有nvcc指令,那就表示没有安装(安装pytorch-gpu中带的cudato

本文标签: 简易版本方法最新Windows