Error with GPU memory in running model

[5hmC]

Hi, I tried to use opus_mut to predict mutation sidechain of protein(86 aa), when finished 3dcnn/csf15/1d_inputs/ss/trr100 and got they in tmp_files, it started running OPUS-Mut, in model1 running, it occupied 23000+M in a 3090(GPU with total 24267M memory) and then failed with gpu in-memory.
The error showed that "no kernel image is available for execution on the device [Op:Tanh]".
If it is convenient, we can exchange emails or wechat.

[thuxugang]

Hi, thanks for your interest.

It is not an OOM error, maybe caused by the version of your cuda enviroment.

The conda env file we used can be found in mut.yml, and the cuda version we used is cuda10.1/cuda10.2.