tensorflow
MAUM
人工智能、数字图像处理、数字信号处理、机械故障诊断
展开
-
python中list和numpy的array的转换
不确定数组的大小,但是迭代存进去的tensor的shape是一样的,那么可以考虑使用下面的方法。 其实我的应用是想申请一个大小不固定的数组,然后每个循环往里面存一次数,最后用于绘制2维图像。 可以使用python内自带的list功能: buf = [] 或者 buf=list([]) 生成一个空的list 然后 for i in range (200): buf.append(...原创 2019-07-03 18:18:42 · 995 阅读 · 0 评论 -
安装TensorFlow-gpu的注意点
直接pip install tensorflow,安装到的是tensorflow-gpu 1.14.0,对应的CUDA版本是10.0,我为啥知道?本以为对应的是最新的CUDA10.1结果安好就悲剧了,还好软件提示:无法找到cudart64_100,请下载CUDA10.0,不过后面给的连接比较坑爹,给的是cuda9.0的,老实的我又中招了,下载了cuda9.0还是用不了。。。后来在下面找到了10.0...原创 2019-06-29 20:12:50 · 1062 阅读 · 0 评论 -
tensorflow实践笔记:屏蔽GPU版tensorflow的提示、分布式训练部署时的问题以及tensorboard的查看
CPU版本的tensorflow代码改到GPU版运行时,会出现很多提示: 上面的提示可以通过加入下面的设置来屏蔽: import os os.environ['TF_CPP_MIN_LOG_LEVEL']='3' 最后的提示可以通过修改tensorflow import时的名字来消除: import tensorflow.compat.v1 as tf 此时运行就是纯净的结果...原创 2019-06-30 08:59:11 · 1259 阅读 · 0 评论