---------------------------------------------------------------------------ImportError Traceback (most recent call last) in
----> 1 import jax.numpy as np
2 from jax import grad, jit, vmap
3 from jax import random
~/external-libraries/jax/__init__.py in
17
18 from jax.version import __version__
---> 19 from jax.api import *
20 from jax import nn
21 from jax import random
~/external-libraries/jax/api.py in
36 from contextlib import contextmanager
37
---> 38 from . import core
39 from . import linear_util as lu
40 from . import ad_util
~/external-libraries/jax/core.py in
28 import numpy as onp
29
---> 30 from . import dtypes
31 from . import linear_util as lu
32
~/external-libraries/jax/dtypes.py in
29 from . import util
30 from .config import flags
---> 31 from .lib import xla_client
32
33 FLAGS = flags.FLAGS
~/external-libraries/jax/lib/__init__.py in
50 except:
51 tpu_client = None
---> 52 from jaxlib import xla_client
53 from jaxlib import lapack
54
~/external-libraries/jaxlib/xla_client.py in
36 # and TensorFlow may fail with duplicate protocol buffer message definitions.
37
---> 38 from . import xla_extension as _xla
39 from .xla_extension import ops
40
ImportError: /home/aistudio/external-libraries/jaxlib/xla_extension.so: symbol cudnnBatchNormalizationForwardTrainingEx, version libcudnn.so.7 not defined in file libcudnn.so.7 with link time reference