使用admet-ai预测

git clone https://github.com/swansonk14/admet_ai
cd admet_ai
micromamba create -n admet_ai python=3.10
micromamba activate admet_ai
# pip install -r requirements.txt
pip install -e .
admet_predict --data_path ./data/molecules.csv --save_path ./data/preds.csv --smiles_column smiles
$env:OMP_NUM_THREADS="12"
$env:MKL_NUM_THREADS="12"
$env:NUMEXPR_NUM_THREADS="12"
$env:PYTORCH_NUM_THREADS="12"
admet_predict --data_path ./data/molecules.csv --save_path ./data/preds.csv --smiles_column smiles --num_workers 0 --no_cache_molecules

pytorch 降级

看起来你遇到的是 PyTorch 2.6+ 把 torch.load 的默认参数改成了 weights_only=True 导致的非兼容问题。admet-ai（通过 chemprop 的 load_args）在加载 checkpoint 里保存的 argparse.Namespace 时没有显式设 weights_only=False，因此触发了 PyTorch 的安全反序列化报错。详见官方说明：2.6 起默认改动与报错指引。

micromamba activate admet_ai
python -c "import torch,sys;print(torch.__version__)"
pip install --upgrade 'torch<2.6'
pip install --upgrade 'torch<2.6' --index-url https://download.pytorch.org/whl/cu121