RuntimeError: expected device cuda:0 and dtype Float but got device cuda:0 and dtype Half

报错如下

在这里插入图片描述

分析原因

可能是pytorch版本问题

解决方法

打开官网

https://pytorch.org/get-started/locally/
在这里插入图片描述

pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

成功安装pytorch1.7.1版本,即可解决此报错!

Logo

北京人形旗下天工造物具身智能开源社区,聚焦具身天工与慧思开物两大平台

更多推荐