Closed WBS-123 closed 4 years ago
Have you compiled the inplaceABN before using the code?
Some parts of InPlace-ABN have a native CUDA implementation, which must be compiled with the following commands:
cd libs
sh build.sh
python build.py
The build.sh
script assumes that the nvcc
compiler is available in the current system search path.
The CUDA kernels are compiled for sm_50
, sm_52
and sm_61
by default.
To change this (e.g. if you are using a Kepler GPU), please edit the CUDA_GENCODE
variable in build.sh
.
I will close this issue. You can contact me if you have any other questions.
@irfanICMLL Hi, I complied the InPlaceABN but got an error "ImportError: torch.utils.ffi is deprecated. Please use cpp extensions instead." when I try to run 'python build.py', the pytorch version is 1.5.0.
from .__ext import lib as _lib, ffi as _ffi i can't find the ._ext/lib and ._ext/ffi this error arise: No module named 'libs._ext._ext' could you fix the problem for me?