Which version do u use ? I have some troubles in torch_scatter

JinheonBaek / RGCN

Pytorch implementation of a RGCN Link Prediction Model

229 stars 41 forks source link

Which version do u use ? I have some troubles in torch_scatter #2

Closed DengYaosheng closed 3 years ago

DengYaosheng commented 4 years ago

It is a great jobs and I'm very appreciate about it. But when i trying to : pip install torch_scatter the terminal returns a stack red error seems that my cuda version not fitted with it So could u please tell me about the version u use ? Thanks a lot~

JinheonBaek commented 4 years ago

Hello, thanks for your interest.

I'm using pytorch 1.4.0 version, and torch_scatter 2.0.4 version. Also, the code was tested on cuda 10.1 environment.