search

ikostrikov / walk_in_the_park

MIT License
241 stars 35 forks source link

Unable to initialize backend #5

Open COST-97 opened 1 year ago

COST-97 commented 1 year ago

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 commented 1 year ago

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