Pytorch无法运行在GPU上,提示显卡版本太低的解决方法

Pytorch无法运行在GPU上,提示显卡版本太低的解决方法


利用下列语句可以让pytorch选择运行在cpu或gpu上:

DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu") 
# 让torch判断是否使用GPU

当提示无法运行在GPU:显卡版本太低。用以下方法解决:

1、检查Pytorch和Cuda的安装情况

用这两个命令可以分别检查pytorch和cuda安装是否成功,返回值如图所示则正常:
在这里插入图片描述
环境的配置可以参考该文章:https://blog.csdn.net/qq_36653505/article/details/83932941。我个人的步骤为:安装Anaconda,单独配置一个装有pytorch的python环境,安装PyCharm,将解释器设置为pytorch环境中的python。

2、升级显卡版本

不建议直接去官网下载,有时候速度慢,有时候会中途断连,当然也可以尝试,有时候比较顺利。
可以使用设备管理器的显示适配器-独显驱动升级来完成,快速稳定。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值