search

jwyang / faster-rcnn.pytorch

A faster pytorch implementation of faster r-cnn
MIT License
7.67k stars 2.33k forks source link

Compiling Issue - Branche: pytorch-1.0 - lib/setup.py #804

Closed stekau closed 4 years ago

stekau commented 4 years ago

I started the setup.py for compilation in the lib direction. The compilation of the code seemed to work fine until a error occured:

4 errors detected in the compilation of "C:/Users/AI/AppData/Local/Temp/tmpxft_000009b0_00000000-10_ROIAlign_cuda.cpp1.ii". error: command 'C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\bin\nvcc.exe' failed with exit status 1

Does anybody have an idea how I can fix the problem? My environment uses the following packages: pytorch 1.2.0 torchvision 0.4.0 cudatoolkit 10.0 python -> 3.7 OS -> Win10

Because of this error I do not get the _C file necessary to train the network as described in #503. As described in #503 I get the following error message:

python trainval_net.py --dataset pascal_voc --net vgg16 --bs 24 --nw 1 --lr 0.001 --lr_decay_step 8 --cuda

Traceback (most recent call last): File "trainval_net.py", line 33, in from model.faster_rcnn.vgg16 import vgg16 File "C:\Users\AI\PycharmProjects\faster-rcnn.pytorch\lib\model\faster_rcnn\vgg16.py", line 16, in from model.faster_rcnn.faster_rcnn import _fasterRCNN File "C:\Users\AI\PycharmProjects\faster-rcnn.pytorch\lib\model\faster_rcnn\faster_rcnn.py", line 10, in from model.rpn.rpn import _RPN File "C:\Users\AI\PycharmProjects\faster-rcnn.pytorch\lib\model\rpn\rpn.py", line 8, in from .proposal_layer import _ProposalLayer File "C:\Users\AI\PycharmProjects\faster-rcnn.pytorch\lib\model\rpn\proposal_layer.py", line 21, in from model.roi_layers import nms File "C:\Users\AI\PycharmProjects\faster-rcnn.pytorch\lib\model\roi_layers__init__.py", line 3, in from .nms import nms File "C:\Users\AI\PycharmProjects\faster-rcnn.pytorch\lib\model\roi_layers\nms.py", line 3, in from model import _C ImportError: cannot import name '_C' from 'model' (C:\Users\Steffen\PycharmProjects\faster-rcnn.pytorch\lib\model__init__.py)

If anybody have some recommendations on this problem I would appreciate. Thanks in advance.

stekau commented 4 years ago

I found a solution for the problem. https://github.com/facebookresearch/maskrcnn-benchmark/issues/254 and https://github.com/facebookresearch/maskrcnn-benchmark/pull/271/files