Closed antran89 closed 5 years ago
Hi, The error looks like occurring during compilation of SigmoidFocalLoss.cu. It is weird because I didn't use any special library here. Are you able to build the main original branch, the [maskrcnn-benchmark] (https://github.com/facebookresearch/maskrcnn-benchmark)?
Thanks @chengyangfu. I solved the issue.
🐛 Bug
I try to build the code inside a NVIDIA docker contaner.
To Reproduce
Steps to reproduce the behavior:
Expected behavior
Should compile the C++ code.
Environment
Please copy and paste the output from the environment collection script from PyTorch (or fill out the checklist below manually).
You can get the script and run it with:
Collecting environment information... PyTorch version: 1.0.1.post2 Is debug build: No CUDA used to build PyTorch: 9.0.176
OS: Ubuntu 16.04.5 LTS GCC version: (Ubuntu 5.4.0-6ubuntu1~16.04.11) 5.4.0 20160609 CMake version: Could not collect
Python version: 3.6 Is CUDA available: Yes CUDA runtime version: 9.0.176 GPU models and configuration: GPU 0: Tesla V100-SXM2-16GB Nvidia driver version: 390.46 cuDNN version: /usr/lib/x86_64-linux-gnu/libcudnn.so.7.2.1
Versions of relevant libraries: [pip] numpy==1.13.3 [pip] torch==1.0.1.post2 [pip] torchsummary==1.5.1 [pip] torchvision==0.2.2.post3 [conda] torch 1.0.1.post2
[conda] torchsummary 1.5.1
[conda] torchvision 0.2.2.post3
conda
,pip
, source):Additional context