$ 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설치
반응형