search

bcgsc / AMPlify

Attentive deep learning model for antimicrobial peptide prediction
Other
36 stars 10 forks source link

Blas xGEMMBatched launch failed #14

Open narib opened 11 months ago

narib commented 11 months ago

Hi I have tried to run AMPlify successfully on CPU only, but slow. Now trying to run AMPlify with tensorflow-gpu (we have 4 x Nvidia A800), it ran into this error consistently in prediction after model loaded.

image

No matter how I tried to set many parameter in session config, e.g. gpu_options.allow_growth =True, (in fact, if false, one GPU mem is not peaked, only one GPU card is used), see below images image PS. my input peptide has two aa sequences only

Being rather new to tensorflow, any chance some one may help advice what 's gone wrong

mark0428 commented 10 months ago

Hi @narib,

Which tensorflow and keras versions are you using?

Cheers, Chenkai