Reference : https://pytorch-geometric.readthedocs.io/en/latest/
PyG Documentation — pytorch_geometric documentation
© Copyright 2023, PyG Team. Revision 688c7541.
pytorch-geometric.readthedocs.io
PyG는 PyTorch에 기반해 만들어진 라이브러리로 구조적 데이터와 관련된 광범위한 어플리케이션들을 위한 GNNs를 쉽게 쓰고 훈련시키는걸 도와줍니다.
출처: https://www.pyg.org/ What is PyG?
이 라이브러리를 처음 설치할 때 버전 확인을 제대로 안해서 한참을 헤매느라 시간을 많이 뺏겼다ㅠㅜ
역시 늘 느끼지만 setting이 제일 어려운것 같아...
Pre-request
Anaconda (PyG를 설치하기 위한 가상환경)
Jupyter Notebook
Installing PyG in virtual env
1. 파이썬 버전 확인
PyG는 Python 3.7~3.10까지 지원한다.
Anaconda Prompt에 들어가서 파이썬의 버전을 확인한다.
$ python -V
나의 경우 Python 3.7.3을 사용하기 때문에 PyG 버전에 부합한다.
만약 3.7~3.10에 해당되지 않는다면 해당되는 버전의 가상환경을 새로 만들어서 그 환경에 설치하자!
관련 이전글
[Anaconda, Jupyter notebook] 파이썬 버전 다른 가상환경과 커널 추가하기
[Anaconda, Jupyter notebook] 파이썬 버전 다른 가상환경과 커널 추가하기
1. Anaconda에 새로운 가상환경 만들기 및 삭제하기 conda create -n [새 가상환경명] python=[원하는 파이썬버전] 원하는 파이썬 버전을 골라 새 가상환경을 만들고 잘 만들어졌는지 리스트를 확인해보자
life-is-potatoo.tistory.com
2. PyG 설치하기
도큐먼트 대로 아나콘다를 통해 명령어를 입력해서 pyg를 설치해준다.
그리고 제대로 설치됐는지 확인해보면
모듈을 찾지 못한다.
그렇게 시작된 나의 삽질....
3. .whl 파일 설치하기
참고링크
https://sweetdev.tistory.com/1130
[DL] torch-geometric 설치하기 (torch-sparse 에러 해결 포함)
일단 가상환경 위 혹은 도커 위에서 설치해야 한다. 공식 문서에서는 root user에서 설치하지 말라고 한다. anaconda로 설치하기 현재 자신의 pytorch 버전이 1.10 이상이고, CUDA버전이 10.2 이상이라면 ana
sweetdev.tistory.com
import torch
print(torch.__version__)
print(torch.version.cuda)위 코드로 torch와 cuda 버전을 확인한다
내 cuda 버전은 None으로 뜨니까 cpu인거다.
https://pytorch-geometric.com/whl/torch-1.10.2%2Bcpu.html
https://pytorch-geometric.com/whl/torch-[torch버전]%2B[cuda버전].html
자기 버전에 맞는 url로 들어가면 다양한 버전의 .whl 파일을 받을 수 있다.
여기서 cp-[해당파이썬버전] 으로 기기의 OS에 맞게 각각 받아주자
pip install [.whl 파일경로]
첨에 잘못 깔렸던게 있어서 설치가 안되길래 추가 옵션으로 --force-reinstall 을 붙여 강제 재설치를 함
이렇게 torch_cluster, torch_scatter, torch_sparse, torch-spline-conv 를 설치해주자
4개를 다 설치해주고 pyg를 import해보면 아무 오류가 뜨지 않는 것을 볼 수 있다.
