qq_33671888的博客

1 ubuntu16.04 cuda10.2 nvidia-driver:470.74 docker镜像 多gpu节点pytorch训练yolov5,出现RuntimeError: cuDNN error: CUDNN_STATUS_INTERNAL_ERROR， 1.1 初步判断是多节点训练引起，，改成单节点，调节batchsize发现小batchsize才不会出错，，但是显卡还没有很多显存，判断是显卡驱动版本引起，因为该镜像在其他机器不会出现这种情况，，把470换成450版本显卡驱动，上面问题消...

yolov5 c++ python java 推理

背景: 因为所里面大多数同学在研究论文时用的比较多的还是pytorch,所以考虑在pc端的部署采用libtorch ,当然 libtorch比较新，所以还是要谨慎地采用,因为看到官方只用resnet等简单的网络做例子时就要有被坑的心里准备，但是，，项目紧急，不管了。。 只要懂得其中一些规则,尤其是jit 的规则还是挺好弄的，但是本人也是第一次真正接触到jit，所以也踩了很多坑 如果熟悉了里...

1 在torchscript 里面使用nn.modulelist时， 例如声明一个 self.vgg= nn.modulelist(....) 不允许使用如 for i in range(len(self.vgg)): y=self.vgg[i](x) 报错:module cannot be used as a value 因为torchscript不能把module当作值对待...