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

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