본문 바로가기
카테고리 없음

torch-scatter설치

by 단창 2022. 6. 17.

$ python -c "import torch; print(torch.__version__)"
>>> 1.8.1

$ python -c "import torch; print(torch.version.cuda)"
>>> 10.1

 

pip install torch-scatter==2.0.8 -f https://data.pyg.org/whl/torch-1.8.1+cu101.html

자신의 torch, cuda 버전에 맞는 torch-scatter설치 

반응형