Error when loading "Preparation" section

[kellynicole6055]

See below

/usr/local/lib/python3.10/dist-packages/kornia/augmentation/augmentation.py:1830: DeprecationWarning: GaussianBlur is no longer maintained and will be removed from the future versions. Please use RandomGaussianBlur instead.
  warnings.warn(
---------------------------------------------------------------------------
OSError                                   Traceback (most recent call last)
<ipython-input-4-33a6c09bcae5> in <cell line: 13>()
     11 
     12 from network.AEI_Net import AEI_Net
---> 13 from coordinate_reg.image_infer import Handler
     14 from insightface_func.face_detect_crop_multi import Face_detect_crop
     15 from arcface_model.iresnet import iresnet100

5 frames
/content/sber-swap/coordinate_reg/image_infer.py in <module>
      2 import numpy as np
      3 import os
----> 4 import mxnet as mx
      5 from skimage import transform as trans
      6 import insightface

/usr/local/lib/python3.10/dist-packages/mxnet/__init__.py in <module>
     21 """MXNet: a concise, fast and flexible framework for deep learning."""
     22 
---> 23 from .context import Context, current_context, cpu, gpu, cpu_pinned
     24 from . import engine, error
     25 from .base import MXNetError

/usr/local/lib/python3.10/dist-packages/mxnet/context.py in <module>
     21 import warnings
     22 import ctypes
---> 23 from .base import classproperty, with_metaclass, _MXClassPropertyMetaClass
     24 from .base import _LIB
     25 from .base import check_call

/usr/local/lib/python3.10/dist-packages/mxnet/base.py in <module>
    354 __version__ = libinfo.__version__
    355 # library instance of mxnet
--> 356 _LIB = _load_lib()
    357 
    358 # type definitions

/usr/local/lib/python3.10/dist-packages/mxnet/base.py in _load_lib()
    345         lib = ctypes.CDLL(lib_path[0], winmode=0x00000008)
    346     else:
--> 347         lib = ctypes.CDLL(lib_path[0], ctypes.RTLD_LOCAL)
    348     # DMatrix functions
    349     lib.MXGetLastError.restype = ctypes.c_char_p

/usr/lib/python3.10/ctypes/__init__.py in __init__(self, name, mode, handle, use_errno, use_last_error, winmode)
    372 
    373         if handle is None:
--> 374             self._handle = _dlopen(self._name, mode)
    375         else:
    376             self._handle = handle

OSError: libcudart.so.11.0: cannot open shared object file: No such file or directory

[luoxyhappy]

Maybe the current CUDA version in Colab is 12.2, and there is no corresponding support for this version in mxnet. You can change !pip install mxnet_cu112 to !pip install mxnet in the third cell. Also, in the "Initialize" cell, modify handler = Handler('./coordinate_reg/model/2d106det', 0, ctx_id=0, det_size=640) to handler = Handler('./coordinate_reg/model/2d106det', 0, ctx_id=-1, det_size=640). This way, it will use the CPU for processing.