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

本節中咱們將展現如何使用多塊 GPU 計算,例如,使用多塊GPU 訓練同一個模型。正如所指望的那樣,運行本節中的程序須要至少2塊 GPU。事實上,一臺機器上安裝多塊 GPU 很常見,這是由於主板上一般會有多個 PCIe 插槽。若是正確安裝了 NVIDIA 驅動,咱們能夠經過nvidia-smi命令來查看當前計算機上的所有 GPU。算法 In [1]: !nvidia-smi   Mon Feb 2
相關文章
相關標籤/搜索