【发布时间】:2021-07-13 19:34:58
【问题描述】:
我正在编写一个脚本,使用xgboost 包在 R 中训练机器学习模型。如果 CUDA GPU 可用并且安装了包的 GPU 版本,则可以加速模型训练。
是否有可以检查 GPU 是否可用的功能(最好不安装除xgboost 之外的其他软件包)?
【问题讨论】:
我正在编写一个脚本,使用xgboost 包在 R 中训练机器学习模型。如果 CUDA GPU 可用并且安装了包的 GPU 版本,则可以加速模型训练。
是否有可以检查 GPU 是否可用的功能(最好不安装除xgboost 之外的其他软件包)?
【问题讨论】:
您可以通过 gce_list_gpus() 查看可用于您的项目和区域的 GPU
试试这个: gce_list_gpus(project="你的项目")
【讨论】:
googleComputeEngineR 包。有没有办法检查 GPU 只使用基本 R 包和xgboost R 包?