首先进行排除:
1.了解自己GPU的型号(任务管理器---性能)(设备管理器---显示适配器)
确定GPU支持cuda
在anaconda prompt中输入命令:
nvidia-smi
是否版本过低,如果过低就去更新下载nvidia驱动程序
import torch
torch.cuda.is_available()
2.下载cuda
选择合适版本安装
3.安装pytorch
选择合适版本之后,复制下方的命令安装
最后再次检测,出现true即可
注意:
版本一定要对应,不然就会一直出错
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)