Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

How to run Complex on GPU colab? #19

Open
Sireesiru opened this issue Apr 4, 2022 · 0 comments
Open

How to run Complex on GPU colab? #19

Sireesiru opened this issue Apr 4, 2022 · 0 comments

Comments

@Sireesiru
Copy link

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?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant