search

malllabiisc / CompGCN

ICLR 2020: Composition-Based Multi-Relational Graph Convolutional Networks
Apache License 2.0
597 stars 107 forks source link

About cuda version #23

Closed yunzi-94 closed 3 years ago

yunzi-94 commented 3 years ago

In your requirement.txt file, torch==1.4.0, torch_scatter==2.0.4. I have installed the version as you said, but it goes wrong:

RuntimeError: Detected that PyTorch and torch_scatter were compiled with different CUDA versions. PyTorch has CUDA version 10.1 and torch_scatter has CUDA version 10.2. Please reinstall the torch_scatter that matches your PyTorch install.

soumyasanyal commented 3 years ago

Hi @yunzi-94, the torch_scatter library installation process has changed from the time we used it. Please refer to the original repo link for details on how to install torch_scatter for the right CUDA version. Specifically, for torch==1.4.0 and CUDA version 10.1 the relevant command should be - pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-1.4.0+cu101.html