1. 首页
  2. 数据库
  3. 其它
  4. Win10 faster rcnn pytorch1.0入坑记录及数据集准备

Win10 faster rcnn pytorch1.0入坑记录及数据集准备

上传者: 2021-01-31 16:30:48上传 PDF文件 71.3KB 热度 64次
环境:win10 py36 cuda10 pytorch1.0 源代码:pytorch1.0 (如果是pytorch0.4可以看pytorch0.4) 两个代码都是linux下的,win10坑很多...... 1、遇到的cl.exe找不到或者.cu导致setup.py编译不成功 网上解决问题很多 ,参考链接:参考1以及参考2 修改setup.py文件及一些ROIPooling等cu文件。 也可以去源码的issue将faster-rcnn.pytorch-pytorch-1.0\lib\model\csrc\cuda中的两个ROIXXXXX.cu进行修改。 要修改的就是这两个.cu文件的dim3 gri
用户评论