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 · 982 阅读 · 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 · 1056 阅读 · 0 评论 -
tensorflow实践笔记:屏蔽GPU版tensorflow的提示、分布式训练部署时的问题以及tensorboard的查看
CPU版本的tensorflow代码改到GPU版运行时,会出现很多提示:上面的提示可以通过加入下面的设置来屏蔽:import osos.environ['TF_CPP_MIN_LOG_LEVEL']='3'最后的提示可以通过修改tensorflow import时的名字来消除:import tensorflow.compat.v1 as tf此时运行就是纯净的结果...原创 2019-06-30 08:59:11 · 1235 阅读 · 0 评论