dgl과 gpu를 사용할 때 에러가 날 경우

Error message 

dgl 관련 코드를 작성할 때 dgl 모듈을 사용하면서 에러가 날 수 있다. 

에러 메시지가 많이 뜨는데 중간에 에러 메시지를 살펴보면 cpu gpu 문제임을 알 수 있다. 

 

 

gpu dgl은 따로 존재한다. 

요약하면 dgl은 cpu버전과 gpu 버전이 따로 있기 때문에 라이브러리를 구별해서 설치해주어야 한다. 

 

만약 cpu dgl을 사용할 예정이라면 

pip install dgl

pip install dgl을 하면 되고 

 

gpu dgl을 사용할 예정이면

pip install dgl-cu110

pip install dgl-cu110

을 설치한다. 

마지막 110은 cuda version에 따라 달라지는데 이는 nvidia-smi 명령어를 통해 확인할 수 있다. 

 

 

nvidia-smi

2022년 12월 기준 nvidia-smi에서 colab notebook은 cuda version 11.2가 설치되어 있다. (우측상단) 

따라서 dgl-cu112가 사실상 맞지만 이 버전의 dgl은 없으므로 cu110을 설치한다. 

 

 

결론 

1. dgl을 사용하다가 오류가 나면 cpu인지 gpu를 사용하고 있는지 확인한다. 

2. cpu면 pip install dgl을 gpu면 pip install dgl-cu110을 설치한다. (코랩 기준)