visualGLM部署过程中遇到的GET was unable to find an engine to execute this computation

torch安装链接参考https://pytorch.org/get-started/previous-versions/

错误如下图所示:
visualGLM部署过程中遇到的GET was unable to find an engine to execute this computation_第1张图片原因猜测是因为torch版本过高,但是torchvision版本过低或者不兼容导致的,所以回退了底一版本的torch2.0.0
之前版本的torch如下图所示:
在这里插入图片描述执行命令完成torch、torchvision、torchaudio降级

conda install pytorch==2.0.0 torchvision==0.15.0 torchaudio==2.0.0 pytorch-cuda=11.8 -c pytorch -c nvidia

visualGLM部署过程中遇到的GET was unable to find an engine to execute this computation_第2张图片visualGLM部署过程中遇到的GET was unable to find an engine to execute this computation_第3张图片如下图所示,只需要不到1.5个G
安装后如下图所示:
在这里插入图片描述之后再执行

python cli_demo.py

便不报错了。

你可能感兴趣的:(pytorch)