Getting cuda initialization error

TimDettmers / ConvE

Convolutional 2D Knowledge Graph Embeddings resources

MIT License

675 stars 163 forks source link

Getting cuda initialization error #36

Closed aayushee closed 5 years ago

aayushee commented 5 years ago

I have installed ConvE earlier and it was working but I recently re-installed it and I am getting following cuda error everytime I run the main.py file. conve

Can someone please help me resolve this error? Also, is it possible to run the embedding model without using cuda, just with pytorch cpu version? I tried that too but it gives me "Torch not compiled with cuda enabled" error.

TimDettmers commented 5 years ago

Thank you, this is actually an issue that needed to be solved. I just pushed a7efed49eb0fe2a89243bc16c13863ee232f551f that removes these imports.