环境搭建如下
镜像：jupyterlab-pytorch:2.3.0-ubuntu22.04-dtk24.04.2-py3.10-devel

git clone https://github.com/chaidiscovery/chai-lab.git

#注释掉 torch 和transformes
pip install -r requirements.in

#chai_lab 需要torch 2.3.1 ，但是社区没有发现此版本，故使用2.3.0
pip install chai_lab

pip install triton==2.1.0

pip install torch-2.3.0+das.opt1.dtk24042-cp310-cp310-manylinux_2_28_x86_64.whl

计算K100Ai报错如下，L20 计算正常
python examples/predict_structure.py

k100ai 计算报错

L20 计算日志