You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Hi
I tried to train my own model using './train_dragnn.sh -v -v &' but it doesn't use GPU device.
I added this code in train_dragnn.py and it is only cpu device available:
from tensorflow.python.client import device_lib
print device_lib.list_local_devices()
Tensorflow compiled with GPU support and when i create new python file i can list devices:
Hi
I tried to train my own model using './train_dragnn.sh -v -v &' but it doesn't use GPU device.
I added this code in train_dragnn.py and it is only cpu device available:
Tensorflow compiled with GPU support and when i create new python file i can list devices:
So, maybe you can suggest me how to use my gpu in training process or maybe i need to put some flags to enable gpu support.
Regards,
Vladimir
The text was updated successfully, but these errors were encountered: