解决“RuntimeError: expected backend CUDA and dtype Float but got backend CUDA and dtype Half”问题

问题描述

在运行Gituhub程序的时候，出现报错

......
y[..., 0:2] = (y[..., 0:2] * 2. - 0.5 + self.grid[i].to(x[i].device)) * self.stride[i]  # xy
RuntimeError: expected backend CUDA and dtype Float but got backend CUDA and dtype Half

解决方法

经过网上搜索相关博客之后，发现可能是Pytorch的版本问题

查询本机的Python版本和Pytorch版本

import torch
print(torch.__version__)

可以看到Python版本为3.6.3，Pytorch版本为1.1.0

查看程序运行所需要的环境

 Python >= 3.7
 Pytorch == 1.5.x

所以需要升级Python和Pytorch的版本，但是如果只是简单的对Python和Pytorch版本进行简单的升级，以后再跑其他程序的时候说不定还需要再次更改版本，非常麻烦；所以建议采用创建虚拟环境的方法，为每一个程序创建一个专属的虚拟环境