AMPlify icon indicating copy to clipboard operation
AMPlify copied to clipboard
verified publisher icon bcgsc

Blas xGEMMBatched launch failed

Open narib opened this issue 2 years ago • 1 comments

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

narib avatar Oct 05 '23 09:10 narib

Hi @narib,

Which tensorflow and keras versions are you using?

Cheers, Chenkai

mark0428 avatar Oct 16 '23 22:10 mark0428