Source code for experiments in the papers "Complex Embeddings for Simple Link Prediction" (ICML 2016) and "Knowledge Graph Completion via Complex Tensor Factorization" (JMLR 2017).
I am currently trying to run Complex on Google Colab Pro using my data.
I have specified the requirements for GPU on Colab and installed all packages in requirements.txt
The algorithm works fine but doesn't use the Colab GPU and uses CPU instead. As a result, I am having to wait long for my output.
Can you please specify what is wrong while using the command : THEANO_FLAGS="device=gpu" & python fb15k_run.py?
I also noticed that it doesn't use all the 15 cores on my Linux system. How can I make sure it is using all the cores?
Hello,
I am currently trying to run Complex on Google Colab Pro using my data. I have specified the requirements for GPU on Colab and installed all packages in requirements.txt The algorithm works fine but doesn't use the Colab GPU and uses CPU instead. As a result, I am having to wait long for my output.
Can you please specify what is wrong while using the command : THEANO_FLAGS="device=gpu" & python fb15k_run.py? I also noticed that it doesn't use all the 15 cores on my Linux system. How can I make sure it is using all the cores?