Running Boltz prediction of protein with 275 residues, 1 ligands C[C@H](CCC(O)=O)[C@H]1CC[C@H]2[C@@H]3CC[C@@H]4C[C@H](O)CC[C@]4(C)[C@H]3CC[C@]12C on gpu
Using cached multiple sequence alignment /home/swlovell/Downloads/ChimeraX/BoltzMSA/TANGO2_lithocholic
Running boltz prediction failed with exit code 1:
command:
/home/swlovell/boltz22/bin/boltz predict /home/swlovell/structures/Gonzalez_Lina/TANGO2/Docking/boltz2/Lithocholic_acid/TANGO2_lithocholic_acid.yaml --use_potentials --accelerator gpu
stdout:
Boltz version 2.2.0
Checking input data.
All inputs are already processed.
Processing 0 inputs with 0 threads.
Running structure prediction for 1 input.
Predicting: | | 0/? [00:00<?, ?it/s] Predicting: 0%| | 0/1 [00:00<?, ?it/s] Predicting DataLoader 0: 0%| | 0/1 [00:00<?, ?it/s] Predicting DataLoader 0: 0%| | 0/1 [00:01<?, ?it/s]
stderr:
0it [00:00, ?it/s] 0it [00:00, ?it/s]
Using bfloat16 Automatic Mixed Precision (AMP)
GPU available: True (cuda), used: True
TPU available: False, using: 0 TPU cores
HPU available: False, using: 0 HPUs
/home/swlovell/boltz22/lib/python3.11/site-packages/pytorch_lightning/trainer/connectors/logger_connector/logger_connector.py:76: Starting from v1.9.0, `tensorboardX` has been removed as a dependency of the `pytorch_lightning` package, due to potential conflicts with other packages in the ML ecosystem. For this reason, `logger=True` will use `CSVLogger` as the default logger, unless the `tensorboard` or `tensorboardX` packages are found. Please `pip install lightning[extra]` or one of them to enable TensorBoard support by default
Fri Aug 7 05:37:27 2026: Loading Boltz structure prediction weights
/home/swlovell/boltz22/lib/python3.11/site-packages/pytorch_lightning/utilities/migration/utils.py:56: The loaded checkpoint was produced with Lightning v2.5.0.post0, which is newer than your current Lightning version: v2.5.0
Fri Aug 7 05:37:38 2026: Finished loading Boltz structure prediction weights
Fri Aug 7 05:37:38 2026: Starting structure inference
You are using a CUDA device ('NVIDIA RTX PRO 4500 Blackwell') that has Tensor Cores. To properly utilize them, you should set `torch.set_float32_matmul_precision('medium' | 'high')` which will trade-off precision for performance. For more details, read https://pytorch.org/docs/stable/generated/torch.set_float32_matmul_precision.html#torch.set_float32_matmul_precision
LOCAL_RANK: 0 - CUDA_VISIBLE_DEVICES: [0]
Fri Aug 7 05:37:39 2026: Begin structure inference TANGO2_lithocholic_acid (1 of 1)
Traceback (most recent call last):
File "/home/swlovell/boltz22/bin/boltz", line 8, in <module>
sys.exit(cli())
^^^^^
File "/home/swlovell/boltz22/lib/python3.11/site-packages/click/core.py", line 1157, in __call__
return self.main(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/swlovell/boltz22/lib/python3.11/site-packages/click/core.py", line 1078, in main
rv = self.invoke(ctx)
^^^^^^^^^^^^^^^^
File "/home/swlovell/boltz22/lib/python3.11/site-packages/click/core.py", line 1688, in invoke
return _process_result(sub_ctx.command.invoke(sub_ctx))
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/swlovell/boltz22/lib/python3.11/site-packages/click/core.py", line 1434, in invoke
return ctx.invoke(self.callback, **ctx.params)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/swlovell/boltz22/lib/python3.11/site-packages/click/core.py", line 783, in invoke
return __callback(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/swlovell/boltz22/lib/python3.11/site-packages/boltz/main.py", line 1355, in predict
trainer.predict(
File "/home/swlovell/boltz22/lib/python3.11/site-packages/pytorch_lightning/trainer/trainer.py", line 859, in predict
return call._call_and_handle_interrupt(
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/swlovell/boltz22/lib/python3.11/site-packages/pytorch_lightning/trainer/call.py", line 47, in _call_and_handle_interrupt
return trainer_fn(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/swlovell/boltz22/lib/python3.11/site-packages/pytorch_lightning/trainer/trainer.py", line 898, in _predict_impl
results = self._run(model, ckpt_path=ckpt_path)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/swlovell/boltz22/lib/python3.11/site-packages/pytorch_lightning/trainer/trainer.py", line 982, in _run
results = self._run_stage()
^^^^^^^^^^^^^^^^^
File "/home/swlovell/boltz22/lib/python3.11/site-packages/pytorch_lightning/trainer/trainer.py", line 1021, in _run_stage
return self.predict_loop.run()
^^^^^^^^^^^^^^^^^^^^^^^
File "/home/swlovell/boltz22/lib/python3.11/site-packages/pytorch_lightning/loops/utilities.py", line 179, in _decorator
return loop_run(self, *args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/swlovell/boltz22/lib/python3.11/site-packages/pytorch_lightning/loops/prediction_loop.py", line 125, in run
self._predict_step(batch, batch_idx, dataloader_idx, dataloader_iter)
File "/home/swlovell/boltz22/lib/python3.11/site-packages/pytorch_lightning/loops/prediction_loop.py", line 255, in _predict_step
predictions = call._call_strategy_hook(trainer, "predict_step", *step_args)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/swlovell/boltz22/lib/python3.11/site-packages/pytorch_lightning/trainer/call.py", line 323, in _call_strategy_hook
output = fn(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^
File "/home/swlovell/boltz22/lib/python3.11/site-packages/pytorch_lightning/strategies/strategy.py", line 438, in predict_step
return self.lightning_module.predict_step(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/swlovell/boltz22/lib/python3.11/site-packages/boltz/model/models/boltz2.py", line 1059, in predict_step
out = self(
^^^^^
File "/home/swlovell/boltz22/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1751, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/swlovell/boltz22/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1762, in _call_impl
return forward_call(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/swlovell/boltz22/lib/python3.11/site-packages/boltz/model/models/boltz2.py", line 464, in forward
z = z + template_module(
^^^^^^^^^^^^^^^^
File "/home/swlovell/boltz22/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1751, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/swlovell/boltz22/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1762, in _call_impl
return forward_call(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/swlovell/boltz22/lib/python3.11/site-packages/boltz/model/modules/trunkv2.py", line 498, in forward
v = v + self.pairformer(v, pair_mask, use_kernels=use_kernels)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/swlovell/boltz22/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1751, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/swlovell/boltz22/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1762, in _call_impl
return forward_call(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/swlovell/boltz22/lib/python3.11/site-packages/boltz/model/layers/pairformer.py", line 329, in forward
z = layer(
^^^^^^
File "/home/swlovell/boltz22/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1751, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/swlovell/boltz22/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1762, in _call_impl
return forward_call(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/swlovell/boltz22/lib/python3.11/site-packages/boltz/model/layers/pairformer.py", line 243, in forward
z = z + dropout * self.tri_mul_out(
^^^^^^^^^^^^^^^^^
File "/home/swlovell/boltz22/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1751, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/swlovell/boltz22/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1762, in _call_impl
return forward_call(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/swlovell/boltz22/lib/python3.11/site-packages/boltz/model/layers/triangular_mult.py", line 92, in forward
return kernel_triangular_mult(
^^^^^^^^^^^^^^^^^^^^^^^
File "/home/swlovell/boltz22/lib/python3.11/site-packages/torch/_dynamo/eval_frame.py", line 838, in _fn
return fn(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^
File "/home/swlovell/boltz22/lib/python3.11/site-packages/boltz/model/layers/triangular_mult.py", line 23, in kernel_triangular_mult
return triangle_multiplicative_update(
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/swlovell/boltz22/lib/python3.11/site-packages/cuequivariance_torch/primitives/triangle.py", line 314, in triangle_multiplicative_update
return f(
^^
File "/home/swlovell/boltz22/lib/python3.11/site-packages/cuequivariance_ops_torch/__init__.py", line 97, in triangle_multiplicative_update
_raise_triton_import_error("triangle_multiplicative_update")
File "/home/swlovell/boltz22/lib/python3.11/site-packages/cuequivariance_ops_torch/__init__.py", line 91, in _raise_triton_import_error
raise Exception(
Exception: Failed to import Triton-based component: triangle_multiplicative_update:
Traceback (most recent call last):
File "/home/swlovell/boltz22/lib/python3.11/site-packages/cuequivariance_ops_torch/attention_pair_bias.py", line 27, in <module>
from torch.fx._symbolic_trace import is_fx_symbolic_tracing
ImportError: cannot import name 'is_fx_symbolic_tracing' from 'torch.fx._symbolic_trace' (/home/swlovell/boltz22/lib/python3.11/site-packages/torch/fx/_symbolic_trace.py)

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
File "/home/swlovell/boltz22/lib/python3.11/site-packages/cuequivariance_ops_torch/__init__.py", line 64, in <module>
from cuequivariance_ops_torch.attention_pair_bias import (
File "/home/swlovell/boltz22/lib/python3.11/site-packages/cuequivariance_ops_torch/attention_pair_bias.py", line 29, in <module>
from torch.fx._symbolic_trace import (
ImportError: cannot import name 'is_fx_tracing_symbolic_tracing' from 'torch.fx._symbolic_trace' (/home/swlovell/boltz22/lib/python3.11/site-packages/torch/fx/_symbolic_trace.py)

Please make sure to install triton==3.3.0. Other versions may not work!
sys:1: DeprecationWarning: builtin type swigvarlink has no __module__ attribute

