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版本，即可解决此报错！