catboost调用gpu就内核崩溃
上个月做一个竞赛,由于cpu实在太慢想着说调用gpu, 在网上看了一些教程以为要安装gpu版本的catboost,后面发现似乎直接把 task_type 参数改为 ‘GPU’ 便可。但是不知道为啥一运行就会出现内核死亡的问题, 在网上简单看了一下, 有的人使用xgboost也会出现这个问题,因为当时比较忙没有来得及解决,在这里标记一下。
·
更多推荐
上个月做一个竞赛,由于cpu实在太慢想着说调用gpu, 在网上看了一些教程以为要安装gpu版本的catboost,后面发现似乎直接把 task_type 参数改为 ‘GPU’ 便可。但是不知道为啥一运行就会出现内核死亡的问题, 在网上简单看了一下, 有的人使用xgboost也会出现这个问题,因为当时比较忙没有来得及解决,在这里标记一下。
更多推荐
所有评论(0)