Skip to content

How to run Complex on GPU colab? #19

Description

@Sireesiru

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?

Activity

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions