原因

个人遇到的问题原因是因为：安装的pytorch为cpu版本；查看pytorch版本类型：

conda list

我这里已经解决了，所以看到的是gpu版本
image.png

解决办法

根源是因为conda默认安装了cpuonly，该模块导致了默认安装的pytorch和tensorflow都会是cpu
版本，因此我们需要将cpuonly模块删除：

conda uninstall cpuonly

使用该命令删除了cpuonly模块后，conda会自动将pytorch转换成gpu版本；如果当前环境下并没有cpuonly模块，那么我们需要先安装该模块：

conda install cpuonly

安装完成后，再使用前面的命令卸载该模块即可：

conda uninstall cpuonly