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.
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