search

AIChallenger / AI_Challenger_2018

AI Challenger, a platform for open datasets and programming competitions to artificial intelligence (AI) talents around the world. https://challenger.ai/
681 stars 280 forks source link

How can I run the code on GPU? #4

Open JustinLin610 opened 6 years ago

JustinLin610 commented 6 years ago

I ran the train_CNN.py and set CUDA_VISIBLE DEVICES, but it still did not run on GPU. In my understanding, keras with tensorflow backend can use GPU automatically.

Safian-Haq commented 5 years ago

I guess specifying which challenge you are talking about can help.