深度学习:如何使用多块 GPU 计算?

本节中我们将展示如何使用多块 GPU 计算,例如,使用多块GPU 训练同一个模型。正如所期望的那样,运行本节中的程序需要至少2块 GPU。事实上,一台机器上安装多块 GPU 很常见,这是因为主板上通常会有多个 PCIe 插槽。如果正确安装了 NVIDIA 驱动,我们可以通过nvidia-smi命令来查看当前计算机上的全部 GPU。 In [1]: !nvidia-smi   Mon Feb 25
相关文章
相关标签/搜索