Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

ImportError on _reorg_layer.so #10

Closed
tymokvo opened this issue Mar 27, 2017 · 4 comments
Closed

ImportError on _reorg_layer.so #10

tymokvo opened this issue Mar 27, 2017 · 4 comments

Comments

@tymokvo
Copy link

tymokvo commented Mar 27, 2017

Hi,

Can't seem to resolve another issue on attempting to run demo.py :

from ._reorg_layer import lib as _lib, ffi as _ffi ImportError: dynamic module does not define init function (init_reorg_layer)

Seems to be thrown on importing the _reorg_layer.so in the /_ext/reorg_layer directory

Using Python 2.7 and Pytorch 0.1.10-py27_1cu80

Also, I'm using Pytorch from within an anaconda environment. Could that be causing problems?

@longcw
Copy link
Owner

longcw commented Mar 29, 2017

Did you run ./make.sh before running the demo?

@tymokvo
Copy link
Author

tymokvo commented Mar 30, 2017

Yeah. I tried recloning the repo and running make.sh again and the ImportError is gone now. So I can close this issue. However, the same file is now throwing cudaCheckError() failed : invalid device function

@longcw longcw closed this as completed Mar 31, 2017
@longcw
Copy link
Owner

longcw commented Mar 31, 2017

See this smallcorgi/Faster-RCNN_TF#19

I got the same error (i.e., cudaCheckError() failed : invalid device function) with my Tesla K40. When I changed the -arch parameter in lib/make.sh to sm_35, and rerun make.sh, it worked.

# Which CUDA capabilities do we want to pre-build for?
# https://developer.nvidia.com/cuda-gpus
#   Compute/shader model   Cards
#   6.1		      P4, P40, Titan X so CUDA_MODEL = 61
#   6.0                    P100 so CUDA_MODEL = 60
#   5.2                    M40
#   3.7                    K80
#   3.5                    K40, K20
#   3.0                    K10, Grid K520 (AWS G2)
#   Other Nvidia shader models should work, but they will require extra startup
#   time as the code is pre-optimized for them.
CUDA_MODELS=30 35 37 52 60 61

@tymokvo
Copy link
Author

tymokvo commented Mar 31, 2017

Yup, making against cuda compute 35 worked. Thanks!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants