when i run cpi_train.py

lishuya17 / MONN

MONN: a Multi-Objective Neural Network for Predicting Pairwise Non-Covalent Interactions and Binding Affinities between Compounds and Proteins

100 stars 32 forks source link

Open t9lex opened 1 year ago

t9lex commented 1 year ago

RuntimeError: cublas runtime error : the GPU program failed to execute at /tmp/pip-req-build-PQwZIe/aten/src/THC/THCBlas.cu:259