PyG 소개 및 설치 방법 (파이썬으로 GNN 모델링하기)

PyG 소개

 PyG는 graph neural network (GNN)를 위한 파이썬 라이브러리로, 그래프 데이터셋을 만드는 것부터 QM9, ZINC 등 다양한 그래프 데이터셋의 다운로드, GCN, GAT 등 다양한 GNN 모델링까지 쉽게 할 수 있게 도와주는 라이브러리이다. 

PyG

 그래프 형태의 데이터는 노드, 엣지, 그리고 그들이 가지고 있는 연결관계, 각각이 가지고 있는 feature 등 다양한 구조를 필요로 하고, 또 이들에 대해서 GNN 모델링을 하기 위해서는 복잡한 구조를 다루어야 하는 경우가 많은데 PyG를 통해서 우리는 이를 쉽게 해볼 수 있다.

PyG 설치 방법

 현재 글을 작성하고 있는 컴퓨터가 Windows 환경이기 때문에, Windows에서 어떻게 PyG를 설치하는지에 대해 알아보겠다. (현재는 PyTorch 1.11. 버전까지 나와 있어서 이를 기준으로 한다. 또한 Anaconda가 설치되어 있다는 것을 가정한다.)

1. Anaconda에 접속해서 본인이 사용할 가상환경을 하나 만든다. 나의 경우 gnn이라는 이름으로 가상환경을 만들었다.

conda create -n gnn

2. 해당 가상환경을 activate해준다.

conda activate gnn

3. 다음 코드를 통해서 pyg를 설치해준다. Proceed?라는 문장이 나오면 y를 입력하면 된다.

conda install pyg -c pyg

열심히 설치하는 모습

4. 잘 설치되었는지 확인을 하기 위해서 설치된 라이브러리에 pyg가 있는지 확인한다. 아래 사진처럼 pyg의 설치한 버전 (나의 경우 2.0.4)이 나오면 성공이다.

conda list | findstr pyg

 

 이번 글에서는 PyG를 어떻게 설치하는지에 대해서 알아보았다. 다음 글부터는 PyG를 어떻게 활용하는지에 대해 하나하나 살펴보려고 한다.

 

References

1. https://pytorch-geometric.readthedocs.io/en/latest/index.html