Unable to initialize backend

[COST-97]

Hello:
There are no problems with the environment configuration process.
But when I run the program, there is a warning, as follows:

I0919 21:58:29.074738 139995721590592 xla_bridge.py:350] Unable to initialize backend 'tpu_driver': NOT_FOUND: Unable to find driver in registry given worker: I0919 21:58:29.120949 139995721590592 xla_bridge.py:350] Unable to initialize backend 'rocm': NOT_FOUND: Could not find registered platform with name: "rocm". Available platform names are: Interpreter CUDA Host I0919 21:58:29.121373 139995721590592 xla_bridge.py:350] Unable to initialize backend 'tpu': module 'jaxlib.xla_extension' has no attribute 'get_tpu_client'

I found that the program cannot call the GPU normally, although tensorflow can call the GPU.

Could you give me some advice?
Thank you!

Ubuntu: 22.04 LTS
NVIDIA Driver Version: 515.65.01
CUDA Version: 11.7
Python: 3.8.13

[ikostrikov]

@COST-97 it seems that it cannot detect cuda. The easiest way to fix it is via installing it with conda.