dyelax / Adversarial_Video_Generation

A TensorFlow Implementation of "Deep Multi-Scale Video Prediction Beyond Mean Square Error" by Mathieu, Couprie & LeCun.
MIT License
736 stars 184 forks source link

question about the gpu #9

Open ghost opened 7 years ago

ghost commented 7 years ago

Hi, dyelax! Thanks for sharing the awesome code!!! It helps me a lot, but I don't know whether the code is using gpu. I don't find any sentence of configuring gpu. When I run the code, I found it didn't use gpu to compute, I don't know why. Please help me! Thank you very much!

dyelax commented 7 years ago

Do you have the GPU version of TensorFlow installed? If so, it should run on GPU automatically. You can find instructions here: https://www.tensorflow.org/install/

Sent from my iPhone

On Apr 22, 2017, at 10:00 PM, GssUCAS notifications@github.com wrote:

Hi, dyelax! Thanks for sharing the awesome code!!! It helps me a lot, but I don't know whether the code is using gpu. I don't find any sentence of configuring gpu. When I run the code, I found it didn't use gpu to compute, I don't know why. Please help me! Thank you very much!

— You are receiving this because you are subscribed to this thread. Reply to this email directly, view it on GitHub, or mute the thread.

ghost commented 7 years ago

Thank for your reply! I have fixed it.