torch.cuda.is

torch.cuda.is,第1张

首先进行排除:

1.了解自己GPU的型号(任务管理器---性能)(设备管理器---显示适配器)

确定GPU支持cuda

在anaconda prompt中输入命令:

nvidia-smi

是否版本过低,如果过低就去更新下载nvidia驱动程序

import torch
torch.cuda.is_available()

2.下载cuda

选择合适版本安装

3.安装pytorch

 选择合适版本之后,复制下方的命令安装

最后再次检测,出现true即可

注意:

版本一定要对应,不然就会一直出错

欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/langs/570438.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-04-09
下一篇 2022-04-09

发表评论

登录后才能评论

评论列表(0条)

保存