RuntimeError: GET was unable to find an engine to execute this computation
因为torch torchvision cuda以及python版本都是有兼容关系的,经过查找材料,原因猜测是因为torch版本过高,但是torchvision版本过低或者不兼容导致的
所以降低版本为torch2.0.0,命令如下
conda install pytorch==2.0.0 torchvision==0.15.0 torchaudio==2.0.0 pytorch-cuda=11.8 -c pytorch -c nvidia
问题解决。