rusty1s / pytorch_scatter

PyTorch Extension Library of Optimized Scatter Operations
https://pytorch-scatter.readthedocs.io
MIT License
1.5k stars 178 forks source link

A problem of install #432

Open xiaosa269 opened 3 months ago

xiaosa269 commented 3 months ago

Hello,

Thank you very much for your pytorch_scatter library. I'm using CUDA: 11.1 and torch: 1.8.1. After installing torch, I conducted some tests, and it seems that the GPU is accessible.

However, after installing pytorch-scatter using conda install pytorch-scatter -c pyg, when I entered Python for testing, the following issue occurred: (frnet) xhy@sun:~/code/FRNet-master$ python Python 3.8.0 (default, Nov 6 2019, 21:49:08) [GCC 7.3.0] :: Anaconda, Inc. on linux Type "help", "copyright", "credits" or "license" for more information.

import torch torch.cuda.is_available() False

After uninstalling pytorch-scatter using conda uninstall pytorch-scatter, I tested torch again: (frnet) xhy@sun:~/code/FRNet-master$ python Python 3.8.0 (default, Nov 6 2019, 21:49:08) [GCC 7.3.0] :: Anaconda, Inc. on linux Type "help", "copyright", "credits" or "license" for more information.

import torch torch.cuda.is_available() Traceback (most recent call last): File "", line 1, in AttributeError: module 'torch' has no attribute 'cuda'

Could you please advise on why this is happening? Your assistance would be greatly appreciated. Thank you very much.

rusty1s commented 3 months ago

Try torch-scatter==2.0.8 for PyTorch 1.8, see here.