先查看安装的pytorch torchvision版本是否为gpu版本,第三列带cpu字眼的则需要从官网下载换成GPU版本的
以上说明是GPU版本,那么问题是因为显卡驱动过低了,更新至最新版本即可,查看显卡驱动和cuda驱动版本
打开cmd:
输入:
cd C:\Program Files\NVIDIA Corporation\NVSMI
nvidia-smi
cuda10.2版本一般需要显卡驱动430以上;利用驱动精灵更新即可;
先查看安装的pytorch torchvision版本是否为gpu版本,第三列带cpu字眼的则需要从官网下载换成GPU版本的
以上说明是GPU版本,那么问题是因为显卡驱动过低了,更新至最新版本即可,查看显卡驱动和cuda驱动版本
打开cmd:
输入:
cd C:\Program Files\NVIDIA Corporation\NVSMI
nvidia-smi
cuda10.2版本一般需要显卡驱动430以上;利用驱动精灵更新即可;