Alibaba-MIIL / ML_Decoder

Official PyTorch implementation of "ML-Decoder: Scalable and Versatile Classification Head" (2021)
MIT License
315 stars 52 forks source link

undefined symbol #75

Closed Q1ngS0ng closed 1 year ago

Q1ngS0ng commented 1 year ago

It's a good paper,but I meet a import error of inplace_abn: (ml) root@21f2f8c38728:~/anaconda3/lib# /root/anaconda3/envs/ml/bin/python /root/ML_Decoder_o/train.py Traceback (most recent call last): File "/root/ML_Decoder_o/train.py", line 12, in from src_files.models import create_model File "/root/ML_Decoder_o/src_files/models/init.py", line 1, in from .utils import create_model File "/root/ML_Decoder_o/src_files/models/utils/init.py", line 1, in from .factory import create_model File "/root/ML_Decoder_o/src_files/models/utils/factory.py", line 11, in from ..tresnet import TResnetM, TResnetL, TResnetXL File "/root/ML_Decoder_o/src_files/models/tresnet/init.py", line 1, in from .tresnet import TResnetM, TResnetL, TResnetXL File "/root/ML_Decoder_o/src_files/models/tresnet/tresnet.py", line 10, in from inplace_abn import InPlaceABN, ABN File "/root/anaconda3/envs/ml/lib/python3.8/site-packages/inplace_abn/init.py", line 1, in from .abn import ABN, InPlaceABN, InPlaceABNSync File "/root/anaconda3/envs/ml/lib/python3.8/site-packages/inplace_abn/abn.py", line 8, in from .functions import inplace_abn, inplace_abn_sync File "/root/anaconda3/envs/ml/lib/python3.8/site-packages/inplace_abn/functions.py", line 8, in from . import _backend ImportError: /root/anaconda3/envs/ml/lib/python3.8/site-packages/inplace_abn/_backend.cpython-38-x86_64-linux-gnu.so: undefined symbol: ZNK2at6Tensor3addEN3c106ScalarES2

This really confuses me, what to do with it? Could u share the detail of python env?

Q1ngS0ng commented 1 year ago

torch 1.9.0 torchvision 0.10.0 nvidia-cublas-cu11 11.10.3.66