Yl

解决pytorch模型加载时gpu id的限制 问题描述 刚开始接触pytorch时，发现每次调用训练好的模型，总是被原来训练时使用的第几个gpu限制。 假如我训练模型时，用的是第3号gpu。那么在测试模型时，加载模型时直接使用GPU的话，就会被限制使用第3号gpu才能运行。 解决方案 首先将模型加载到cpu，然后再使用GPU。

问题描述 有时在加载已训练好的模型时，会出现 out of memory 的错误提示，但仔细检测使用的GPU卡并没有再用且内存也没有超出。 经查阅发现原来是训练模型时使用的GPU卡和加载时使用的GPU卡不一样导致的。个人感觉，因为pytorch的模型中是会记录有GPU信息的，所以有时使用不同的GPU加载时会报错。 解决方法 gpu之间的相互转换。即，将训练时的gpu卡转换为加载时的g...

问题描述 用pytorch加载训练好的模型时，可能会遇到下面的问题： AttributeError: 'module' object has no attribute '_rebuild_tensor_v2' 原因 由于训练模型时使用的是新版本的pytorch，而加载时使用的是旧版本的pytorch。 解决方法 1、升级pytorch到适当的新版本 2、或者，在程序开头添加下面的代...

pytorch报错 RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 1 and 3 in dimension 1 at /pytorch/aten/src/TH/generic/THTensorMath.cpp:3616 原因分析 使用DataLoader加载图...