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

在这里插入图片描述

问题解决。

Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐