如何启用英特尔 iRIS Xe GPU 进行深度学习？

Posted 2023-03-16

【中文标题】如何启用英特尔 iRIS Xe GPU 进行深度学习？

【英文标题】：How to enable intel iRIS Xe GPU for Deep Learning?

【发布时间】：2021-11-12 13:21:16

【问题描述】：

当我使用 google colab 训练数据以及将运行时切换到 GPU 时。我的 colab 只使用 CPU 而不是 GPU。如何启用使用 Windows 10 系统的 Intel GPU？

【参考方案1】：

要使您的笔记本能够使用 GPU 运行时，请选择运行时 >“更改运行时类型”菜单，然后从硬件加速器下拉菜单中选择 GPU。

然后您可以通过在您的笔记本单元之一中运行以下代码来确保：

gpu_info = !nvidia-smi
gpu_info = '\n'.join(gpu_info)
if gpu_info.find('failed') >= 0:
  print('Select the Runtime > "Change runtime type" menu to enable a GPU accelerator, ')
  print('and then re-execute this cell.')
else:
  print(gpu_info)