- 博客(4)
- 收藏
- 关注
原创 mmdetection环境搭建
会有两个文件() rtmdet_tiny_8xb32-300e_coco.py 和 rtmdet_tiny_8xb32-300e_coco_20220902_112414-78e30dcc.pth)下载到你的工程文件夹中。推理后的结果会放入outputs文件夹中,在 outputs/vis 文件夹中有包含网络预测的检测框的图片demo.jpg。安装mmcv,注意mmcv的版本要和pytorch和cuda的版本匹配,可通过官网查看。先cd到mmdetection程序文件的根目录下再安装相关依赖。
2024-09-22 16:03:44 281
原创 RuntimeError: NCCL error in: /pytorch/torch/lib/c10d/ProcessGroupNCCL.cpp:825, invalid usage
nccl报错表明在使用DistributedDataParallel ()分布式训练时出现了问题。问题解决,程序可运行!
2024-07-09 20:51:12 581
原创 RuntimeError:Input type(torch.cuda.HalfTensor)and weight type(torch.FloatTensor)should be the same
报错分析:输入数据张量类型为torch.cuda.HalfTensor,已经加载在GPU上,而添加的模块的权重参数张量类型为torch.FloatTensor),添加的模块在CPU上,没有在GPU上,两者不匹配。
2024-05-31 16:10:34 707
原创 Could not load library libcudnn_ops_infer.so.8. Error: libcudnn_ops_infer.so.8
中,如果有libcudnn_ops_infer.so.8文件,该文件应该是在/home/lab/anaconda3/envs/xxx/lib/python3.8/site-packages/nvidia/cudnn/lib路径下。将其中的/home/lab/anaconda3/envs/xxx/lib/python3.8/site-packages/nvidia/cudnn/lib替换为自己的路径。再重新运行程序即可,问题解决!
2024-05-16 10:49:25 720 3
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人