How do i fix the "Unable to initialize backend 'tpu_driver': NOT_FOUND: Unable to find driver in registry given worker:" error

I0608 08:48:12.491065 140245639700480 run_docker.py:116] Mounting /home/aditya/Documents/Spodoptera_frugiperda/proteins -> /mnt/fasta_path_0
I0608 08:48:12.491170 140245639700480 run_docker.py:116] Mounting /home/aditya/Documents/Spodoptera_frugiperda/proteins -> /mnt/fasta_path_1
I0608 08:48:12.491236 140245639700480 run_docker.py:116] Mounting /home/aditya/Documents/Spodoptera_frugiperda/proteins -> /mnt/fasta_path_2
I0608 08:48:12.491298 140245639700480 run_docker.py:116] Mounting /home/aditya/Documents/Spodoptera_frugiperda/proteins -> /mnt/fasta_path_3
I0608 08:48:12.491358 140245639700480 run_docker.py:116] Mounting /home/aditya/Documents/Spodoptera_frugiperda/proteins -> /mnt/fasta_path_4
I0608 08:48:12.491417 140245639700480 run_docker.py:116] Mounting /home/aditya/Documents/Spodoptera_frugiperda/proteins -> /mnt/fasta_path_5
I0608 08:48:12.491476 140245639700480 run_docker.py:116] Mounting /home/aditya/Documents/Spodoptera_frugiperda/proteins -> /mnt/fasta_path_6
I0608 08:48:12.491533 140245639700480 run_docker.py:116] Mounting /home/aditya/Documents/Spodoptera_frugiperda/proteins -> /mnt/fasta_path_7
I0608 08:48:12.491591 140245639700480 run_docker.py:116] Mounting /home/aditya/Documents/Spodoptera_frugiperda/proteins -> /mnt/fasta_path_8
I0608 08:48:12.516574 140245639700480 run_docker.py:116] Mounting /media/aditya/New Volume/protein_database/uniref90 -> /mnt/uniref90_database_path
I0608 08:48:12.516844 140245639700480 run_docker.py:116] Mounting /media/aditya/New Volume/protein_database/mgnify -> /mnt/mgnify_database_path
I0608 08:48:12.516937 140245639700480 run_docker.py:116] Mounting /media/aditya/New Volume/protein_database -> /mnt/data_dir
I0608 08:48:12.525529 140245639700480 run_docker.py:116] Mounting /media/aditya/New Volume/protein_database/pdb_mmcif/mmcif_files -> /mnt/template_mmcif_dir
I0608 08:48:12.536713 140245639700480 run_docker.py:116] Mounting /media/aditya/New Volume/protein_database/pdb_mmcif -> /mnt/obsolete_pdbs_path
I0608 08:48:12.541885 140245639700480 run_docker.py:116] Mounting /media/aditya/New Volume/protein_database/pdb70 -> /mnt/pdb70_database_path
I0608 08:48:12.552317 140245639700480 run_docker.py:116] Mounting /media/aditya/New Volume/protein_database/small_bfd -> /mnt/small_bfd_database_path
I0608 08:48:19.318056 140245639700480 run_docker.py:258] I0608 03:18:19.316803 139978587400000 templates.py:857] Using precomputed obsolete pdbs /mnt/obsolete_pdbs_path/obsolete.dat.
I0608 08:48:21.933797 140245639700480 run_docker.py:258] I0608 03:18:21.933366 139978587400000 xla_bridge.py:353] Unable to initialize backend 'tpu_driver': NOT_FOUND: Unable to find driver in registry given worker:
I0608 08:48:22.052647 140245639700480 run_docker.py:258] I0608 03:18:22.052219 139978587400000 xla_bridge.py:353] Unable to initialize backend 'rocm': NOT_FOUND: Could not find registered platform with name: "rocm". Available platform names are: Interpreter Host CUDA
I0608 08:48:22.052834 140245639700480 run_docker.py:258] I0608 03:18:22.052553 139978587400000 xla_bridge.py:353] Unable to initialize backend 'tpu': module 'jaxlib.xla_extension' has no attribute 'get_tpu_client'
I0608 08:48:22.052919 140245639700480 run_docker.py:258] I0608 03:18:22.052629 139978587400000 xla_bridge.py:353] Unable to initialize backend 'plugin': xla_extension has no attributes named get_plugin_device_client. Compile TensorFlow with //tensorflow/compiler/xla/python:enable_plugin_device set to true (defaults to false) to enable this.
I0608 08:48:31.968616 140245639700480 run_docker.py:258] I0608 03:18:31.968227 139978587400000 run_alphafold.py:466] Have 5 models: ['model_1_pred_0', 'model_2_pred_0', 'model_3_pred_0', 'model_4_pred_0', 'model_5_pred_0']
I0608 08:48:31.968794 140245639700480 run_docker.py:258] I0608 03:18:31.968350 139978587400000 run_alphafold.py:480] Using random seed 1666331041262993896 for the data pipeline
I0608 08:48:31.968888 140245639700480 run_docker.py:258] I0608 03:18:31.968505 139978587400000 run_alphafold.py:218] Predicting XP_035454815.2
I0608 08:48:31.970123 140245639700480 run_docker.py:258] I0608 03:18:31.969789 139978587400000 jackhmmer.py:133] Launching subprocess "/usr/bin/jackhmmer -o /dev/null -A /tmp/tmpxxi5lneh/output.sto --noali --F1 0.0005 --F2 5e-05 --F3 5e-07 --incE 0.0001 -E 0.0001 --cpu 8 -N 1 /mnt/fasta_path_0/XP_035454815.2.fasta /mnt/uniref90_database_path/uniref90.fasta"
I0608 08:48:31.991148 140245639700480 run_docker.py:258] I0608 03:18:31.990779 139978587400000 utils.py:36] Started Jackhmmer (uniref90.fasta) query