
深度学习
钓鱼小队长
这个作者很懒,什么都没留下…
展开
-
Windows10下VS2017+caffe2
平台:Windows10_x64工具:VS2017,anaconda3(python3.6),Cmake 这主要是记录我遇到的一些错误,完整的教程请先看后面那三个教程。 在GitHub下载caffe2源码:https://github.com/caffe2/caffe2,最好是clone,不要下载.zip会缺少包,可以用VS下载。然后打开caffe2/scripts,用命令...原创 2018-03-13 19:34:08 · 10290 阅读 · 2 评论 -
ubuntu16.04多版本cuda8+cudnn5.1和cuda9+cudnn7安装
此方法是在cuda8+cudnn5.1是在安装好的基础下进行的,cuda8+cudnn5.1的安装方法请搜索其他人的方法,在此不多介绍。 环境:ubuntu16.04,GTX1080 Ti,驱动nvidia-384,已装cuda8+cudnn5.1,两个账户aaa,bbb(避免cuda软链切换) cuda9下载可以从英特尔官网下载(runfile)或者到此博客[1]的百度云连接...原创 2018-06-03 10:13:05 · 1091 阅读 · 0 评论 -
tensorflow 释放图,删除变量
问题,在一个程序内构建好了一个图,运行完之后想重新使用这个图进行计算,或者想同时在train完的时候做test,就会提示***变量已存在。解决办法:待一个图运行完之后加上tf.reset_default_graph来重置默认的图。...原创 2018-08-02 17:00:52 · 10591 阅读 · 2 评论 -
TensorFlow中获取大于零的元素集合
a为tensoridx = tf.where(a > 0)output = tf.gather_nd(a, idx)原创 2018-08-13 17:34:17 · 9423 阅读 · 0 评论