查看tensorflow是否使用GPU 

import tensorflow as tf
tf.config.list_physical_devices('GPU')

 

posted @ 2020-03-09 19:31  airfy  阅读(5845)  评论(0)    收藏  举报