Pytorch-geometric Installation

Source: Deep Learning on Medium

Pytorch-geometric Installation

因應未來研究可能需要GNN/GCN之類的技術,所以過年期間試著安裝pytorch-geometric,歷經了一番折騰……(以為要用sudo,結果向學弟請教卻發現不用Orz)

但之前在趕實驗時,用學弟提供的 requirements.txt安裝包含pytorch-geometric的套件,卻讓我過年花大把時間好不容易得來的成果毀於一旦……(想不通為什麼會有undefined symbol:的error……Orz)

後來去官網看,似乎資訊又更新了……按照步驟安裝後,終於又裝回來了XD

以下是(根據官網指示)成功安裝的指令:

pip(3) install --verbose --no-cache-dir torch-scatter==latest+${CUDA} torch-sparse==latest+${CUDA} -f https://s3.eu-central-1.amazonaws.com/pytorch-geometric.com/whl/torch-1.4.0.html
pip(3) install torch-cluster
pip(3) install torch-spline-conv
pip(3) install torch-geometric

注意:

  1. ${CUDA}=92/100/101,分別對應至CUDA 9.2、10.0、10.1
  2. 需有Pytorch 1.4
  3. Python3使用者可用pip3

期待Pytorch-geometric對自己未來研究會有相當的幫助!

若大家有其他安裝/使用 Pytorch/Pytorch-geometric上的心得,歡迎留言討論指教,謝謝!