본문 바로가기

데이터 과학/딥러닝 FrameWork

torch_scatter 설치

본인의 경우 python3.8버전에 pytorch는 1.12.1 버전이었고, cuda는 11.6 버전이었다.

rusty1s/pytorch_scatter: PyTorch Extension Library of Optimized Scatter Operations (github.com) 에서

Binaries로 설치하기 위해 https://data.pyg.org/whl/  사이트에서 아래의 설치 파일을 다운 받아 pip로 설치 하였다.

하지만,, 계속 에러가 났다.

 

 

(이것 때문에 시간을 1시간 30분 낭비하다가...)

 

pytorch를 1.12.0으로 다운그레이드 해주었고(Previous PyTorch Versions | PyTorch), cuda는 11.6 버전 유지.

pip install torch==1.12.0+cu116 torchvision==0.13.0+cu116 torchaudio==0.12.0 --extra-index-url https://download.pytorch.org/whl/cu116

 

torch-1.12.0 버전의 whl 파일로 다시 설치하였다.

 

설치가 완료 되었다.

import torch_scatter

 

torch 버전을 한단계만 다운그레이드 하니 설치가 되는 허무함이 발생 했다.

반응형