How to run the demo with GPU about deepctr HOT 5 CLOSED

@Cong-Zou
first check if you have instelled deepctr and tensorflow-gpu in the same evironment.
then, run the following code and make sure thers is at least one GPU device in the output of the code.

import os
from tensorflow.python.client import device_lib
os.environ["TF_CPP_MIN_LOG_LEVEL"] = "99"
 
if __name__ == "__main__":
    print(device_lib.list_local_devices())

from deepctr.

@Cong-Zou Hi,if you have installed tensorflow-gpu,it will run with GPU automatically

from deepctr.

Thank you for the reply..
The fact is that I have installed tensorflow-gpu 1.6.0..
But it still runs with CPU..

from deepctr.

They are in the same environment for sure.
Thanks a lot！
I will test the code tomorrow morning and tell you the result then..

from deepctr.

I have solved it. Thank you!
I uninstalled the tensorflow-gpu-1.6.0 and deepctr and then I installed them again. Now it finally works.
The reason might be that my tensorflow-gpu has some package dependency problems..

from deepctr.