自定义博客皮肤VIP专享

*博客头图:

格式为PNG、JPG,宽度*高度大于1920*100像素,不超过2MB,主视觉建议放在右侧,请参照线上博客头图

请上传大于1920*100像素的图片!

博客底图:

图片格式为PNG、JPG,不超过1MB,可上下左右平铺至整个背景

栏目图:

图片格式为PNG、JPG,图片宽度*高度为300*38像素,不超过0.5MB

主标题颜色:

RGB颜色,例如:#AFAFAF

Hover:

RGB颜色,例如:#AFAFAF

副标题颜色:

RGB颜色,例如:#AFAFAF

自定义博客皮肤

-+
  • 博客(13)
  • 收藏
  • 关注

原创 进程已死,显存还在被占用(无fuser命令)

由于某种原因,进程挂掉了,但占用的显存未释放。情况:服务器内无fuser,7卡被占用,nvidia-smi显示无对应进程。网上的方法大多都是采用fuser查看,但服务器上没有fuser,也没有权限,网上其他方法也不work,然后自己试出来这个方法。解决:显示所有进程( ps aux|grep python ),找到占用显存的进程并杀死( kill -9 PID )。注:如果不知道是哪些进程占用,可以计算下对应显存大小。如果是自己引起的,直接把自己名下的进程

2020-11-03 16:12:45 18

原创 RuntimeError: CUDA error: device-side assert triggered

RuntimeError: CUDA error: device-side assert triggered 断言形式和下述类似::operator()(int)->auto: block: [0,0,0], thread: [4,0,0] Assertion index >= -sizes[i] && index < sizes[i] && “index out of bounds” failed./opt/conda/conda-bld/pytorc

2020-11-03 08:47:18 21

原创 pytorch 模型参数(卷积)转不到cuda中,RuntimeError: Input type (torch.cuda.FloatTensor) and weight type

pytorch 模型参数转不到cuda中报错:RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the错误示例:class my_layer(nn.Module): def __init__(self, in_channels, out_channels, num_layer): super(my_layer, self).__init_

2020-09-11 15:40:21 87

原创 pytorch load模型参数(.pth),RuntimeError: version_ <= kMaxSupportedFileFormatVersion INTERNAL ASSERT FAIL

pytorch load模型参数(.pth),RuntimeError: version_ <= kMaxSupportedFileFormatVersion INTERNAL ASSERT FAIL原因:使用较高版本的pytorch训练得到的模型,在低版本的ptorch中load时,存在的版本问题。我使用1.6.0版本训练的模型,在1.4.0中是load不了的。解决:升级版本或者使用低版本训练。要不找个一种兼容的保持和读取方法。...

2020-08-13 11:41:44 1124 1

原创 在服务器linux中切换nvcc版本

在服务器linux中切换nvcc版本在nvidia-smi 中cuda版本在conda中nvcc版本这里显示当前使用的nvcc版本由于conda中nvcc版本低于要求,所以要更换。直接去/usr/local 中看服务器安装的cuda版本,然后使用10.1,通过命令行vim ~/.bashrc修改conda的cuda版本然后sourcesource ~/.bashrc结果:...

2020-07-29 10:43:31 118

原创 python 对相同数据进行相同计算,结果不同

python 对相同数据进行相同计算,结果不同这种情况现发现在CPU中存在,在GPU中不存在。暂认为是处理时采用的数据精度不同

2020-07-22 21:08:33 104

原创 Pytorch在forward中创建了新的tensor时,RuntimeError: arguments are located on different GPUs

Pytorch在forward中创建了新的tensor时,RuntimeError: arguments are located on different GPUs这个问题浪费了我一天时间,找了很多关于相关错误提示的解决方案。很不幸,有用但解决不了。下面进入正题:我的错误引起是:我在自定义forward中声明了一个tensorclass test(nn.Module): def __init__(self, xxxx): super(test, self).__init__

2020-07-17 17:07:20 90

原创 pytorch自定义操作层,正向和反向

torch 自定义操作层, 前向和反向过程在torch中定义正向一定要注意是否使用Variable比如,在定义层的forward中正确定义是这样的: def forward(self, infeature): infeature = infeature.reshape(self.inshape[1], self.inshape[2]) / self.p_pixel_num infeature = torch.mm(infeature, self.pooling)

2020-06-30 09:39:56 102

原创 python 数据中存在nan

python 数据中显示nannan的表示这个数是个非数,就像 1除0.所以检查下哪里存在除0的情况,或者有其他的位置存在计算成非数。nan会随着计算传播至后续计算

2020-06-29 17:21:56 312

原创 解决AttributeError

解决AttributeError: object has no attribute ‘log_softmax’错误例子:net = FCN(data)直接在构建网络时,给与输入改为:net = FCN()out = net(data)

2020-06-16 19:42:51 171

原创 ONNX生成模型遇到的问题

在使用onnx生成模型遇到的问题onnx.onnx_cpp2py_export.checker.ValidationError: Nodes in a graph must be topologically sorted, however input ‘ends’ of node:input: “x3” input: “starts” input: “ends” output: “Y” op_...

2019-11-06 22:25:40 1530 1

原创 图像贴图(旋转缩放平移)

自己写的一个贴图算法,虽然很简单,很容易就实现了,但还是要记录一下其中过程。其中有个矩阵运算,其实是这样写的。x2 = (int)(exa_x1cos(rot) - exa_y1sin(rot)) ;//旋转后坐标y2 = (int)(exa_y1cos(rot) + exa_x1sin(rot)) ;身为小白,自己实现了这个简单的算法,也是很高兴的。希望对其他人也有所帮助。rz和cz...

2019-07-19 11:01:56 144

原创 基于皮肤分割的磨皮算法

查阅了博客上绝大多数关于皮肤分割和磨皮美颜的算法,然后自己找了各位博主的代码,有自己将各部分代码组合修改调试,最终写了份参照代码。至于引用哪些博主的代码,找的博客太多分不清楚了,有博主看到引用代码可以私聊我加上你的引用。 我感觉做出来的效果不太好,不如别人算法效果好。 此代码只是做了皮肤分割和滤波磨皮,美白的代码还没写。还有就是发布文章我不太懂它的分类选项,只好随便选了。发布文章的规则我也不...

2019-07-16 16:35:20 311

空空如也

空空如也

空空如也

TA创建的收藏夹 TA关注的收藏夹

TA关注的人 TA的粉丝

提示
确定要删除当前文章?
取消 删除