데이터 과학/딥러닝 FrameWork
torch_scatter 설치
EastHoon
2022. 12. 16. 17:28
본인의 경우 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 버전을 한단계만 다운그레이드 하니 설치가 되는 허무함이 발생 했다.
반응형