본인의 경우 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 버전을 한단계만 다운그레이드 하니 설치가 되는 허무함이 발생 했다.
반응형
'데이터 과학 > 딥러닝 FrameWork' 카테고리의 다른 글
Tensorboard에서 Open3D 사용하기 (0) | 2023.02.19 |
---|---|
RNN & LSTM 설명 및 구현(pytorch) (0) | 2021.03.29 |
모델 앙상블(ensemble) 하기 (0) | 2021.02.26 |
Trouble Shooting (0) | 2021.01.31 |
Torch 데이터셋 & 데이터 로더 + Transforms (0) | 2021.01.30 |