- 博客(3)
- 收藏
- 关注
原创 记录下pytorch安装中一个比较隐蔽的坑,torch.cuda.is_available()返回false的一个原因
如题,想要安装GPU版,一开始从网上找了教程,显卡驱动、CUDA、cuDNN挨个更新安装,都选择了10.1版本,安装完后在anaconda中按照官网提示输入下面的代码,可以看到cudatoolkit即GPU版 conda install pytorch torchvision cudatoolkit=10.1 -c pytorch 然后发现基本上大一点的包都下载不下来,包括pytorch、cudatoolkit等等,遂又从网上找教程,加清华源,加了几个之后下载很快,下载下来也可以用,以为已经成功了,但是t
2020-07-04 00:47:37 2793 5
原创 对多维数组计算中axis的理解
对多维矩阵计算中axis的理解 以tf.reduce_mean()为例 axis参数用来指定操作轴,对于二维矩阵来说是比较好理解的,比如上面的求均值函数,axis=0为每行的对应元素求均值,axis=1为每列的对应元素求均值。但是个人感觉这样记忆一是容易记混,二是难以拓展到高维。所以想到了一种理解方式,供大家参考。 我这种理解其实就是对于上面加粗的对应元素的扩展,二维时候是axis=0是每行的第一个元素求均值、第二个元素……,axis=1是每列的第一个元素求均值,第二个元素…… 扩展到多维其实也是一样,我把
2020-05-28 18:14:08 576
原创 Selective Search论文阅读笔记
小白的计算机视觉学习之路 刚一开始接触到这个领域感觉挺费解的,硬着头皮看论文,看到的就是这个人针对这个问题引入了这个结构,然后效果非常好,但是这其中的因果关系完全没有头绪,更别提自己实现了。 但作为这个方向的研究生,还是要不可避免地踏上这条路,长假在家希望能写出一些东西帮帮跟我一样迷茫的朋友们 从CNN开始 近几年的计算机视觉肯定是离不开CNN,是基础中的基础,一开始我的理解就是卷积核在一幅图上...
2020-02-20 11:39:15 387
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人