GraphLearn-for-PyTorch (GLT) 是一個基於PyTorch的圖學習庫,目的是使分佈式GNN訓練和推理變得簡單和高效。
GLT利用GPU加速圖採樣,並利用UVA來減少頂點和邊特徵的轉換和復制。對於大規模圖,它支持通過高效的分佈式採樣和特徵查找在多個GPU或多台機器上進行分佈式訓練。此外,GLT提供了靈活的分佈式訓練部署,以滿足不同的需求。
GTL 架構如下:
#GraphLearnforPyTorch首頁文檔和下載 #基於PyTorch #的圖學習庫 #程式开发
GraphLearn-for-PyTorch (GLT) 是一個基於PyTorch的圖學習庫,目的是使分佈式GNN訓練和推理變得簡單和高效。
GLT利用GPU加速圖採樣,並利用UVA來減少頂點和邊特徵的轉換和復制。對於大規模圖,它支持通過高效的分佈式採樣和特徵查找在多個GPU或多台機器上進行分佈式訓練。此外,GLT提供了靈活的分佈式訓練部署,以滿足不同的需求。
GTL 架構如下:
#GraphLearnforPyTorch首頁文檔和下載 #基於PyTorch #的圖學習庫 #程式开发