筆記-深度學習模型訓練：在Colab上安裝DGL

前言

最近在研究GAT，在網路上看到使用torch和DGL實作的GAT模型的程式碼，就想說下載下來自己跑跑看，這篇文章：Understand Graph Attention Network。途中遇到問題，把找到的解法記錄下來，給也有一樣問題的朋友參考。

正文

!pip​ install dgl

會出現下面錯誤：

FileNotFoundError: Cannot find DGL C++ graphbolt library

參考網路上資訊：FileNotFoundError: Cannot find DGL C++ graphbolt library at ...，將torch降版至2.1.0，會出現另一個錯誤：

ModuleNotFoundError: No module named 'torch.utils._import_utils'

這個錯誤在網路上能查到的資訊不多，讓我一度卡關。後來直接搜尋"如何在Colab安裝及引用DGL"，看到這篇文章：dgl torch incompatability error: Cannot load Graphbolt C++ library，依照文章下面的建議這麼做：

!pip install torch==2.3.0
!pip install dgl -f https://data.dgl.ai/wheels/torch-2.3/repo.html

import os

os.environ['DGLBACKEND'] = "pytorch"

就可以順利地安裝和引用DGL了。

參考

• Understand Graph Attention Network
• FileNotFoundError: Cannot find DGL C++ graphbolt library at ...
• dgl torch incompatability error: Cannot load Graphbolt C++ library
• ChatGPT

小結

覺得自己對很多東西還不是很清楚，例如：pip、cuda、conda、GNN等，所以要：繼續趕路，繼續留腳印(XD)，哈哈，週末愉快！