Results 2 comments of yushu

Try mmagic docker ? or maybe it cause by pytorch 2.x version try 1.x `conda install pytorch=1.10`