deep-burst-sr icon indicating copy to clipboard operation
deep-burst-sr copied to clipboard

MAGMA not initialised during training

Open wallematthias opened this issue 3 years ago • 3 comments

Hi,

when I train the dbsr model on the default_realworld setting I get the following error during training:

Error in magma getdevice arch: MAGMA not initialized (call magma init() first) or bad device First, I thought this was due to multiprocessing start method, but it is already set to spawn.

Any Ideas?

Best, Matthias

wallematthias avatar Oct 20 '21 08:10 wallematthias

What's your pytorch version? When my torch version==1.9.0, I got the same MAGMA error. And I fixed the problem by downgrading pytorch to 1.7 version.

yjsunnn avatar Oct 20 '21 08:10 yjsunnn

Hi susan,

You are correct I had 1.9 installed, downgrading to the following solved the problem:

# CUDA 10.2
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.2 -c pytorch

THANK YOU :)

wallematthias avatar Oct 20 '21 08:10 wallematthias

Hi,

Thanks for raising this issue @wallematthias and for the solution @susan-sun1999. I will check this and post a note on the install page.

goutamgmb avatar Oct 21 '21 09:10 goutamgmb