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版本,即可解决此报错!
更多推荐
所有评论(0)