Win10 faster rcnn pytorch1.0入坑记录及数据集准备
环境: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
下载地址
用户评论