环境: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
2021-10-13 14:24:26
71KB
AS
c
cnn
1