
Deep Learning
wardenjohn
从事云计算、操作系统方向研究
展开
-
显式调用了cuda以后,pytorch的张量还是没有转移带GPU上的解决方法
在使用GPU加速计算的时候,有的时候将张量显式调用了以后会发现其实还是在cpu上计算打个比方:class A: def __init__(self): super(...) a = torch.randn((1,2,3)).cuda()ORclass B: def __init(self): a = torch.randn((1,2,3))b = B()b.cuda()在显式调用了cuda以后.原创 2020-07-01 14:42:57 · 2328 阅读 · 1 评论 -
Kears 使用:通过回调函数保存最佳准确率下的模型
1:首先,我给我的MixTest文件夹里面分好了类的图片进行重命名(因为分类的时候没有注意导致命名有点不好)def load_data(path): Rename the picture [a tool] for eachone in os.listdir(path): newname = eachone[7:] os.rename(path...原创 2018-11-13 16:57:41 · 2637 阅读 · 0 评论 -
Keras 使用:加载已经训练好的模型进行预测
使用Keras训练好的模型用来直接进行预测,这个时候我们该怎么做呢?【我这里使用的就是一个图片分类网络】现在让我来说说怎么样使用已经训练好的模型来进行预测判定把首先,我们已经又有了model模型,这个模型被保存为model.h5文件然后我们需要在代码里面进行加载model = load_model("model.h5")假设我们自己已经写好了一个load_data函数【loa...原创 2018-11-14 11:20:35 · 45381 阅读 · 41 评论 -
nsync_cv.h :找不到该文件
目前在打算跑faster rcnn的一个GitHub项目在make的时候发现出现了不少的问题现在来说说nsync_cv.h:no such file的问题吧我去找了这个为的解决方案,但是我解决问题的方法跟别人的还是有点不同的首先,当出现这个问题的时候,terminal会报错,你按照报错的提示给出了路径,到达该路径指示的地方然后将#include "nsync_cv.h"...原创 2018-11-30 14:48:28 · 828 阅读 · 0 评论 -
Ubuntu 16.04 安装CUDA9.0后出现驱动不工作且nividia-smi无法链接到NVIDIA驱动的解决办法
首先,遇到的样的问题说明你应该是已经装好了cuda了但是在ubuntu16.04下,你会不会发现安装好了cuda了以后,屏幕的分辨率很蛋疼,而且无法调整?那可能是在你安装NVIDIA驱动的时候出了问题在终端里面输入nvidia-smi发现报错:unable connect to nvidia driver这个真的是个坑,我重装了多次都是没有用:现在解决了首先,如果安装好了c...原创 2018-12-10 11:22:35 · 5687 阅读 · 1 评论