深度學習:如何使用多塊 GPU 計算?

本節中我們將展示如何使用多塊 GPU 計算,例如,使用多塊GPU 訓練同一個模型。正如所期望的那樣,運行本節中的程序需要至少2塊 GPU。事實上,一臺機器上安裝多塊 GPU 很常見,這是因爲主板上通常會有多個 PCIe 插槽。如果正確安裝了 NVIDIA 驅動,我們可以通過nvidia-smi命令來查看當前計算機上的全部 GPU。 In [1]: !nvidia-smi   Mon Feb 25
相關文章
相關標籤/搜索