如何检查CUDA GPU是否在R中可用



我正在编写一个脚本,该脚本使用xgboost包在R中训练机器学习模型。如果CUDA GPU可用并且安装了该包的GPU版本,则可以加速模型训练。

是否有一个功能可以检查GPU是否可用(最好不安装xgboost以外的附加包(?

您可以通过gce_list_GPUs((查看可用于您的项目和区域的GPU

试试这个:gce_list_gpus(项目="您的项目"(

最新更新