2020年8月12日
tensorflow2 前向传播DEMO
摘要： import tensorflow as tffrom tensorflow import kerasfrom tensorflow.keras import datasetsimport osos.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'# x: [60k, 28 阅读全文
posted @ 2020-08-12 10:09 kpwong 阅读(208) 评论(0) 推荐(0) 编辑
怎么查看keras 或者 tensorflow 正在使用的GPU
摘要： from keras import backend as K K.tensorflow_backend._get_available_gpus() from tensorflow.python.client import device_lib import tensorflow as tf prin 阅读全文
posted @ 2020-08-12 09:11 kpwong 阅读(620) 评论(0) 推荐(0) 编辑