我正在编写一个脚本,该脚本使用xgboost
包在R中训练机器学习模型。如果CUDA GPU可用并且安装了该包的GPU版本,则可以加速模型训练。
是否有一个功能可以检查GPU是否可用(最好不安装xgboost
以外的附加包(?
您可以通过gce_list_GPUs((查看可用于您的项目和区域的GPU
试试这个:gce_list_gpus(项目="您的项目"(