2020年2月26日
获取tensorflow中变量的值
摘要： names=[i.name for i in tf.all_variables()]for i in names: print i ker=tf.get_default_graph().get_tensor_by_name("conv1/kernel:0") 阅读全文
posted @ 2020-02-26 18:49 吃饱就睡 阅读(931) 评论(0) 推荐(0)
tensorflow开发环境版本组合
摘要： 记录下各模块的版本 tensorflow 1.15.0 print tf.__version__ cuda 10.0.130 nvcc -v cudnn 7.6.4 cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2 python 阅读全文
posted @ 2020-02-26 00:54 吃饱就睡 阅读(152) 评论(0) 推荐(0)
tensorflow打印可用设备列表
摘要： from tensorflow.python.client import device_libprint(device_lib.list_local_devices()) 阅读全文
posted @ 2020-02-26 00:43 吃饱就睡 阅读(414) 评论(0) 推荐(0)