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

本节中咱们将展现如何使用多块 GPU 计算,例如,使用多块GPU 训练同一个模型。正如所指望的那样,运行本节中的程序须要至少2块 GPU。事实上,一台机器上安装多块 GPU 很常见,这是由于主板上一般会有多个 PCIe 插槽。若是正确安装了 NVIDIA 驱动,咱们能够经过nvidia-smi命令来查看当前计算机上的所有 GPU。算法 In [1]: !nvidia-smi   Mon Feb 2
相关文章
相关标签/搜索