When I use torch_scatter, I get error as followed here.
RuntimeError: Detected that PyTorch and torch_scatter were compiled with different CUDA versions. PyTocatter has CUDA version 10.1. Please reinstall the torch_scatter that matches your PyTorch install.
My torch vision is torch==1.8.0+cu111 and torch-scatter’s is torch-scatter==2.0.6 , how can I get right torch-scatter
You would have to specify the same CUDA version (in your case CUDA11.1) as described in the README of pytorch_scatter
.
Thanks a lot, sir
I have tried
pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-1.8.0+cu111.html
Now torch-scatter==2.0.7, but there is still the same error.
Since it’s the same error I assume pytorch-geometric
still claims to be using CUDA10.1?
In that case, your install command wasn’t successful and you would have to uninstall the previous package and reinstall the new one.